Test against Keras 3

Author: mattdangerw

.github/workflows/actions.yml

@@ -65,6 +65,7 @@ jobs:
     - name: Install dependencies
       run: |
           pip install -r requirements.txt --progress-bar off
+          pip install keras-nightly -U --progress-bar off
           pip install --no-deps -e "." --progress-bar off
     - name: Test with pytest
       env:

requirements-jax-cuda.txt

@@ -1,5 +1,6 @@
 # Tensorflow cpu-only version.
-tensorflow>=2.14.0
+tf-nightly==2.15.0.dev20231009  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.15.0.dev20231009  # Pin a working nightly until rc0.
 tensorflow-text>=2.14.0
 
 # Torch cpu-only version.

requirements-tensorflow-cuda.txt

@@ -1,6 +1,7 @@
 # Tensorflow with cuda support.
-tensorflow[and-cuda]>=2.14.0
-tensorflow-text>=2.14.0
+--extra-index-url https://pypi.nvidia.com
+tf-nightly[and-cuda]==2.15.0.dev20231009  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.15.0.dev20231009  # Pin a working nightly until rc0.
 
 # Torch cpu-only version.
 --extra-index-url https://download.pytorch.org/whl/cpu

requirements-torch-cuda.txt

@@ -1,6 +1,6 @@
 # Tensorflow cpu-only version.
-tensorflow>=2.14.0
-tensorflow-text>=2.14.0
+tf-nightly==2.15.0.dev20231009  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.15.0.dev20231009  # Pin a working nightly until rc0.
 
 # Torch with cuda support.
 --extra-index-url https://download.pytorch.org/whl/cu118

requirements.txt

@@ -1,6 +1,6 @@
 # Tensorflow.
-tensorflow>=2.14.0
-tensorflow-text>=2.14.0
+tf-nightly==2.15.0.dev20231009  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.15.0.dev20231009  # Pin a working nightly until rc0.
 
 # Torch.
 --extra-index-url https://download.pytorch.org/whl/cpu