Pin tensorflow-text to go with tensorflow (#22362)

* Pin tensorflow-text to go with tensorflow * Make it more convenient to pin TensorFlow * setup don't like f-strings
2023-03-24 10:54:06 -04:00
parent 01203475c9
commit 6587125c0a
3 changed files with 4 additions and 3 deletions
--- a/setup.py
+++ b/setup.py
@@ -164,9 +164,10 @@ _deps = [
    "starlette",
    "sudachipy>=0.6.6",
    "sudachidict_core>=20220729",
+    # TensorFlow pin. When changing this value, update examples/tensorflow/_tests_requirements.txt accordingly
    "tensorflow-cpu>=2.4,<2.12",
    "tensorflow>=2.4,<2.12",
-    "tensorflow-text",
+    "tensorflow-text<2.12",
    "tf2onnx",
    "timeout-decorator",
    "timm",