Update

mraveri · Sep 22, 2024 · 42e369a · 42e369a
1 parent 19a0d49
commit 42e369a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -26,6 +26,10 @@ jobs:
             tensorflow-version: "2.15.0"
           - tensorflow-version: "2.15.0"
             tfp-version: "0.24.0"
+          - tensorflow-version: "2.16.1"
+            tfp-version: "0.23.0"
+          - tensorflow-version: "2.17.0"
+            tfp-version: "0.23.0"
 
     steps:
 
@@ -40,11 +44,11 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install tensorflow==${{ matrix.tensorflow-version }}
+          pip install tensorflow-probability==${{ matrix.tfp-version }}
           if ${{ matrix.tensorflow-version }} >= "2.16.0"; then
             pip install tf-keras
             export TF_USE_LEGACY_KERAS=1
           fi
-          pip install tensorflow-probability==${{ matrix.tfp-version }}
           pip install coveralls
           pip install -r requirements.txt