test

Author: mattdangerw

.github/workflows/actions.yml

@@ -7,7 +7,7 @@ on:
     types: [created]
 jobs:
   build:
-    name: Test the code with tf.keras
+    name: Test the code with Keras 2
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v2
@@ -29,7 +29,8 @@ jobs:
             ${{ runner.os }}-pip-
       - name: Install dependencies
         run: |
-          pip install -r requirements.txt --progress-bar off
+          pip install -r requirements-common.txt --progress-bar off
+          pip install tensorflow-text keras-core
           pip install --no-deps -e "." --progress-bar off
       - name: Test with pytest
         run: |
@@ -38,7 +39,7 @@ jobs:
         run: |
           python pip_build.py --install && cd integration_tests && pytest .
   multibackend:
-    name: Test the code with Keras Core
+    name: Test the code with Keras 3
     strategy:
       fail-fast: false
       matrix:
@@ -65,7 +66,6 @@ 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:

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
 

requirements-common.txt

@@ -1,5 +1,4 @@
 # Library deps.
-keras-core>=0.1.6
 dm-tree
 regex
 rich
@@ -17,5 +16,3 @@ namex
 rouge-score
 sentencepiece
 tensorflow-datasets
-# Breakage fix.
-ml-dtypes==0.2.0

requirements-jax-cuda.txt

@@ -1,7 +1,6 @@
 # Tensorflow cpu-only version.
-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
+tf-nightly==2.16.0.dev202310024  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.16.0.dev202310024  # Pin a working nightly until rc0.
 
 # Torch cpu-only version.
 --extra-index-url https://download.pytorch.org/whl/cpu

requirements-tensorflow-cuda.txt

@@ -1,7 +1,7 @@
 # Tensorflow with cuda support.
 --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.
+tf-nightly[and-cuda]==2.16.0.dev20231024  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.16.0.dev20231024  # 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.
-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.
+tf-nightly-cpu==2.16.0.dev20231024  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.16.0.dev20231024  # 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.
-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.
+tf-nightly==2.16.0.dev20231024  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.16.0.dev20231024  # Pin a working nightly until rc0.
 
 # Torch.
 --extra-index-url https://download.pytorch.org/whl/cpu