Support loading keras 3 nightly (#1286)

The nightly version will actually be consider to have a value less that "3.0.0" by python packaging, as the dev prefix indicates a pre-release.
keras-team · Oct 26, 2023 · bbb4b1e · bbb4b1e
1 parent aff79b3
commit bbb4b1e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_nlp/backend/config.py b/keras_nlp/backend/config.py
@@ -66,7 +66,7 @@
     _MULTI_BACKEND = True
 
 # If keras is version 3, use multi-backend keras (our only option).
-_IS_KERAS_3 = version.parse(keras.__version__) >= version.parse("3.0.0")
+_IS_KERAS_3 = version.parse(keras.__version__) >= version.parse("3.0.0.dev0")
 if _IS_KERAS_3:
     _MULTI_BACKEND = True