Pin TF until tests are fixed (#14283)

* Pin TF until tests are fixed * Also pin TF CPU
2021-11-04 21:15:42 -04:00
parent b90a48f654
commit a14d62b0b1
2 changed files with 4 additions and 4 deletions
--- a/setup.py
+++ b/setup.py
@@ -151,8 +151,8 @@ _deps = [
    "sphinxext-opengraph==0.4.1",
    "sphinx-intl",
    "starlette",
-    "tensorflow-cpu>=2.3",
-    "tensorflow>=2.3",
+    "tensorflow-cpu>=2.3,<2.7",
+    "tensorflow>=2.3,<2.7",
    "timeout-decorator",
    "timm",
    "tokenizers>=0.10.1,<0.11",