Temporary use older version of Keras to fix the build

comet-ml · Jul 8, 2024 · 5c79886 · 5c79886
1 parent 37159db
commit 5c79886
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integrations/model-training/ray-train/notebooks/Comet_with_ray_train_keras.ipynb b/integrations/model-training/ray-train/notebooks/Comet_with_ray_train_keras.ipynb
@@ -41,7 +41,7 @@
    },
    "outputs": [],
    "source": [
-    "%pip install comet_ml \"ray[air]>=2.1.0\" tensorflow"
+    "%pip install -U comet_ml \"ray[air]>=2.1.0\" \"keras<3\" \"tensorflow<2.16.0\""
    ]
   },
   {