Added tf 2.18 tests

mraveri · Oct 26, 2024 · 8227e5c · 8227e5c
1 parent e2c632f
commit 8227e5c
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -19,7 +19,7 @@ jobs:
       matrix:
 
         python-version: ["3.9", "3.10", "3.12"]
-        tensorflow-version: ["2.15.0", "2.16.1", "2.17.0"]
+        tensorflow-version: ["2.15.0", "2.16.1", "2.17.0", "2.18.0"]
         tfp-version: ["0.23.0", "0.24.0"]
 
         exclude:
@@ -31,6 +31,8 @@ jobs:
             tfp-version: "0.23.0"
           - tensorflow-version: "2.17.0"
             tfp-version: "0.23.0"
+          - tensorflow-version: "2.18.0"
+            tfp-version: "0.23.0"
 
     steps:
 
@@ -46,6 +48,7 @@ jobs:
           python -m pip install --upgrade pip
           pip install tensorflow==${{ matrix.tensorflow-version }}
           pip install tensorflow-probability==${{ matrix.tfp-version }}
+
           # Install tf-keras for tensorflow 2.16.1
           if [ "${{ matrix.tensorflow-version }}" == "2.16.1" ]; then
             pip install tf-keras
@@ -54,6 +57,11 @@ jobs:
           if [ "${{ matrix.tensorflow-version }}" == "2.17.0" ]; then
             pip install tf-keras
           fi
+          # Install tf-keras for tensorflow 2.18.0
+          if [ "${{ matrix.tensorflow-version }}" == "2.17.0" ]; then
+            pip install tf-keras
+          fi
+
           pip install coveralls
           pip install -r requirements.txt
 
@@ -75,6 +83,9 @@ jobs:
           if [ "${{ matrix.tensorflow-version }}" == "2.17.0" ]; then
             export TF_USE_LEGACY_KERAS=1
           fi
+          if [ "${{ matrix.tensorflow-version }}" == "2.18.0" ]; then
+            export TF_USE_LEGACY_KERAS=1
+          fi
           make test_with_coverage
 
       - name: Upload coverage to coveralls