Update

mraveri · Sep 22, 2024 · 32a26c2 · 32a26c2
1 parent 16e415d
commit 32a26c2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -69,7 +69,12 @@ jobs:
 
       - name: Run unit tests
         run: |
-          export TF_USE_LEGACY_KERAS=1
+          if [ "${{ matrix.tensorflow-version }}" == "2.16.1" ]; then
+            export TF_USE_LEGACY_KERAS=1
+          fi
+          if [ "${{ matrix.tensorflow-version }}" == "2.17.0" ]; then
+            export TF_USE_LEGACY_KERAS=1
+          fi
           make test_with_coverage
 
       - name: Upload coverage to coveralls