Add a test against keras-nightly

keras-team · Jul 12, 2024 · e9c68ec · e9c68ec
1 parent 360da5b
commit e9c68ec
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/.github/workflows/actions.yml b/.github/workflows/actions.yml
@@ -17,10 +17,12 @@ jobs:
       fail-fast: false
       matrix:
         backend: [tensorflow, jax, torch]
-        version: [latest]
+        version: [stable]
         include:
-          - backend: torch
+          - backend: jax
             version: 3.1
+          - backend: jax
+            version: nightly
     runs-on: ubuntu-latest
     env:
       KERAS_BACKEND: ${{ matrix.backend }}
@@ -46,11 +48,16 @@ jobs:
       run: |
           pip install -r requirements.txt --progress-bar off
           pip install --no-deps -e "." --progress-bar off
-    - name: Pin Keras version
+    - name: Pin Keras 3.1
       if: ${{ matrix.version == '3.1'}}
       run: |
         pip uninstall -y keras
         pip install keras==3.1.0 --progress-bar off
+    - name: Pin Keras Nightly
+      if: ${{ matrix.version == 'nightly'}}
+      run: |
+        pip uninstall -y keras
+        pip install keras-nightly --progress-bar off
     - name: Test with pytest
       run: |
         pytest keras_nlp/