From 30e343862f0f78512faae0374fc39898bd00ce95 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Mon, 11 May 2020 21:03:30 -0400 Subject: [PATCH] pin TF to 2.1 (#4297) * pin TF to 2.1 * Pin flake8 as well --- setup.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 873d643ea6..f0f9af8bf5 100644 --- a/setup.py +++ b/setup.py @@ -67,8 +67,8 @@ extras = {} extras["mecab"] = ["mecab-python3"] extras["sklearn"] = ["scikit-learn"] -extras["tf"] = ["tensorflow"] -extras["tf-cpu"] = ["tensorflow-cpu"] +extras["tf"] = ["tensorflow <= 2.1"] +extras["tf-cpu"] = ["tensorflow-cpu <= 2.1"] extras["torch"] = ["torch"] extras["serving"] = ["pydantic", "uvicorn", "fastapi", "starlette"] @@ -79,9 +79,9 @@ extras["docs"] = ["recommonmark", "sphinx", "sphinx-markdown-tables", "sphinx-rt extras["quality"] = [ "black", "isort @ git+git://github.com/timothycrosley/isort.git@e63ae06ec7d70b06df9e528357650281a3d3ec22#egg=isort", - "flake8", + "flake8==3.7.9", ] -extras["dev"] = extras["testing"] + extras["quality"] + ["mecab-python3", "scikit-learn", "tensorflow", "torch"] +extras["dev"] = extras["testing"] + extras["quality"] + ["mecab-python3", "scikit-learn", "tensorflow <= 2.1", "torch"] setup( name="transformers",