set upper limit for keras

Keras 3 introduced a number of backwards incompatible changes that make some dependencies (shap, rise) fail.
dianna-ai · Sep 19, 2024 · 03b0254 · 03b0254
1 parent 8af438c
commit 03b0254
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -52,6 +52,8 @@ install_requires =
     torch <=2.3.1  # this is temporary, see #829
     # tf and tfprob are required but not declared by onnx_tf
     tensorflow_probability[tf]
+    # some dependencies are still incompatible with keras 3 ((kernel)shap, rise)
+    keras<3
     # https://setuptools.pypa.io/en/latest/userguide/datafiles.html#accessing-data-files-at-runtime
     importlib_resources;python_version<'3.10'