Pin Keras for now (#26904)

* Pin Keras for now out of paranoia * Add the keras pin to _tests_requirements.txt too * Make sure the Keras version matches the TF one * make fixup
2023-10-19 14:39:31 +01:00
parent 73dc23f786
commit cbd278f0f6
3 changed files with 4 additions and 0 deletions
--- a/setup.py
+++ b/setup.py
@@ -125,6 +125,8 @@ _deps = [
    "jaxlib>=0.4.1,<=0.4.13",
    "jieba",
    "kenlm",
+    # Keras pin - this is to make sure Keras 3 doesn't destroy us. Remove or change when we have proper support.
+    "keras<2.15",
    "keras-nlp>=0.3.1",
    "librosa",
    "nltk",