From 570a4bb147b64adb3a4b9cb9e183a00d4cb11582 Mon Sep 17 00:00:00 2001 From: Matt Watson Date: Tue, 17 Oct 2023 14:18:23 -0700 Subject: [PATCH] Test against Keras 3 --- .github/workflows/actions.yml | 1 + requirements-jax-cuda.txt | 3 ++- requirements-tensorflow-cuda.txt | 5 +++-- requirements-torch-cuda.txt | 4 ++-- requirements.txt | 4 ++-- 5 files changed, 10 insertions(+), 7 deletions(-) diff --git a/.github/workflows/actions.yml b/.github/workflows/actions.yml index 7d4a7d4ed0..9a35b0b8ec 100644 --- 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 index bb115b14f8..54c4b4b58c 100644 --- 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 index 4b2cf167ea..4390590484 100644 --- 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 index 14e94dd862..3ab18fd24b 100644 --- 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 index aa289402fd..fa502ba37a 100644 --- 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