lock keras version (#5809)

BUG
tensorflow · Nov 3, 2021 · c16a217 · c16a217
1 parent b989f6e
commit c16a217
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tfjs-converter/python/BUILD b/tfjs-converter/python/BUILD
@@ -48,6 +48,7 @@ py_wheel(
     license = "Apache 2.0",
     python_tag = "py3",
     requires = [
+        "keras<=2.6.0",
         "tensorflow-estimator<=2.6.0",
         "tensorflow>=2.1.0,<3",
         "six>=1.12.0,<2",

diff --git a/tfjs-converter/python/requirements.txt b/tfjs-converter/python/requirements.txt
@@ -1,3 +1,4 @@
+keras<=2.6.0
 tensorflow-estimator<=2.6.0
 tensorflow>=2.1.0,<3
 protobuf==3.17.3; python_version < "3"