From 9d0d3a6645384e236c55d311f3f8b7dd67d58562 Mon Sep 17 00:00:00 2001 From: sgugger Date: Mon, 27 Jul 2020 18:03:09 -0400 Subject: [PATCH] Pin TF while we wait for a fix --- setup.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index c4fc91ab317..2ea0954b425 100644 --- a/setup.py +++ b/setup.py @@ -70,14 +70,14 @@ extras["sklearn"] = ["scikit-learn"] # keras2onnx and onnxconverter-common version is specific through a commit until 1.7.0 lands on pypi extras["tf"] = [ - "tensorflow", + "tensorflow<=2.2", # "onnxconverter-common", # "keras2onnx" "onnxconverter-common @ git+git://github.com/microsoft/onnxconverter-common.git@f64ca15989b6dc95a1f3507ff6e4c395ba12dff5#egg=onnxconverter-common", "keras2onnx @ git+git://github.com/onnx/keras-onnx.git@cbdc75cb950b16db7f0a67be96a278f8d2953b48#egg=keras2onnx" ] extras["tf-cpu"] = [ - "tensorflow-cpu", + "tensorflow-cpu<=2.2", # "onnxconverter-common", # "keras2onnx" "onnxconverter-common @ git+git://github.com/microsoft/onnxconverter-common.git@f64ca15989b6dc95a1f3507ff6e4c395ba12dff5#egg=onnxconverter-common", @@ -86,7 +86,7 @@ extras["tf-cpu"] = [ extras["torch"] = ["torch"] extras["serving"] = ["pydantic", "uvicorn", "fastapi", "starlette"] -extras["all"] = extras["serving"] + ["tensorflow", "torch"] +extras["all"] = extras["serving"] + ["tensorflow<=2.2", "torch"] extras["testing"] = ["pytest", "pytest-xdist", "timeout-decorator", "psutil"] # sphinx-rtd-theme==0.5.0 introduced big changes in the style. @@ -97,7 +97,7 @@ extras["quality"] = [ "isort @ git+git://github.com/timothycrosley/isort.git@e63ae06ec7d70b06df9e528357650281a3d3ec22#egg=isort", "flake8", ] -extras["dev"] = extras["testing"] + extras["quality"] + ["mecab-python3<1", "scikit-learn", "tensorflow", "torch"] +extras["dev"] = extras["testing"] + extras["quality"] + ["mecab-python3<1", "scikit-learn", "tensorflow<=2.2", "torch"] setup( name="transformers",