SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Pin Keras cause they messed their release (#14262)

Browse Source 
* Pin Keras cause they messed their release

* Put != instead of <

* Try this way

* Back to the beginning but more agressive
This commit is contained in:
Sylvain Gugger
2021-11-03 15:03:09 -04:00
committed by GitHub
parent 1149243184
commit a1c15ea855
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
setup.py
View File

@@ -107,6 +107,7 @@ _deps = [
"jax>=0.2.8", "jax>=0.2.8",
"jaxlib>=0.1.65", "jaxlib>=0.1.65",
"jieba", "jieba",
"keras!=2.7.0", # Remove when they fix their release
"keras2onnx", "keras2onnx",
"nltk", "nltk",
"numpy>=1.17", "numpy>=1.17",
@@ -227,8 +228,8 @@ extras = {}
extras["ja"] = deps_list("fugashi", "ipadic", "unidic_lite", "unidic") extras["ja"] = deps_list("fugashi", "ipadic", "unidic_lite", "unidic")
extras["sklearn"] = deps_list("scikit-learn") extras["sklearn"] = deps_list("scikit-learn")
extras["tf"] = deps_list("tensorflow", "onnxconverter-common", "keras2onnx") extras["tf"] = deps_list("tensorflow", "onnxconverter-common", "keras2onnx", "keras")
extras["tf-cpu"] = deps_list("tensorflow-cpu", "onnxconverter-common", "keras2onnx") extras["tf-cpu"] = deps_list("tensorflow-cpu", "onnxconverter-common", "keras2onnx", "keras")
extras["torch"] = deps_list("torch") extras["torch"] = deps_list("torch")

1
src/transformers/dependency_versions_table.py
View File

@@ -25,6 +25,7 @@ deps = {
"jax": "jax>=0.2.8", "jax": "jax>=0.2.8",
"jaxlib": "jaxlib>=0.1.65", "jaxlib": "jaxlib>=0.1.65",
"jieba": "jieba", "jieba": "jieba",
"keras": "keras!=2.7.0",
"keras2onnx": "keras2onnx", "keras2onnx": "keras2onnx",
"nltk": "nltk", "nltk": "nltk",
"numpy": "numpy>=1.17", "numpy": "numpy>=1.17",