Test against Keras 3

keras-team · Oct 18, 2023 · 570a4bb · 570a4bb
1 parent 07e1cc2
commit 570a4bb
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 7 deletions.
diff --git a/.github/workflows/actions.yml b/.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:

diff --git a/requirements-jax-cuda.txt b/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.

diff --git a/requirements-tensorflow-cuda.txt b/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

diff --git a/requirements-torch-cuda.txt b/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

diff --git a/requirements.txt b/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