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
This commit is contained in:
Sylvain Gugger 2023-03-24 10:54:06 -04:00 committed by GitHub
parent 01203475c9
commit 6587125c0a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 3 deletions

View File

@ -1,4 +1,4 @@
tensorflow<2.11
tensorflow<2.12
tensorboard
scikit-learn
seqeval

View File

@ -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",

View File

@ -72,7 +72,7 @@ deps = {
"sudachidict_core": "sudachidict_core>=20220729",
"tensorflow-cpu": "tensorflow-cpu>=2.4,<2.12",
"tensorflow": "tensorflow>=2.4,<2.12",
"tensorflow-text": "tensorflow-text",
"tensorflow-text": "tensorflow-text<2.12",
"tf2onnx": "tf2onnx",
"timeout-decorator": "timeout-decorator",
"timm": "timm",