Update

mraveri · Sep 22, 2024 · e8d8541 · e8d8541
1 parent 8f644c2
commit e8d8541
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -56,6 +56,8 @@ jobs:
           python -c "import sys; print(sys.version)"
           python -c "import tensorflow as tf; print(tf.__version__)"
           python -c "import tensorflow_probability as tfp; print(tfp.__version__)"
+          python -c "import tf_keras as tfk; print(tfk.__version__)"
+          echo "TF_USE_LEGACY_KERAS: $TF_USE_LEGACY_KERAS"
 
       - name: Run unit tests
         run: |