also install tf-keras for TF 2.15 as it also checks TF_USE_LEGACY_KERAS

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Feb 27, 2024 · 4af3e1a · 4af3e1a
1 parent 455ae8f
commit 4af3e1a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/find_tensorflow.py b/backend/find_tensorflow.py
@@ -137,7 +137,7 @@ def get_tf_requirement(tf_version: str = "") -> dict:
     if not (tf_version == "" or tf_version in SpecifierSet(">=2.12", prereleases=True)):
         extra_requires.append("protobuf<3.20")
     # keras 3 is not compatible with tf.compat.v1
-    if tf_version == "" or tf_version in SpecifierSet(">=2.16.0rc0", prereleases=True):
+    if tf_version == "" or tf_version in SpecifierSet(">=2.15.0rc0", prereleases=True):
         extra_requires.append("tf-keras; python_version>='3.9'")
         # only TF>=2.16 is compatible with Python 3.12
         extra_requires.append("tf-keras>=2.16.0rc0; python_version>='3.12'")