pin TF to 2.1 (#4297)

* pin TF to 2.1 * Pin flake8 as well

pin TF to 2.1 (#4297)
* pin TF to 2.1 * Pin flake8 as well
30e34386 · Lysandre Debut · GitHub · 56e8ef63 · 30e34386
Unverified Commit 30e34386 authored May 11, 2020 by Lysandre Debut Committed by GitHub May 11, 2020
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

setup.py setup.py +4 -4

No files found.
--- 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",