Ensure that the correct version of keras is installed for TensorFlow …

…CI (#1846)
PennyLaneAI · Nov 3, 2021 · 997d0a2 · 997d0a2
1 parent 01ec130
commit 997d0a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -51,7 +51,7 @@ jobs:
 
       - name: Conditionally install TensorFlow
         if: contains(matrix.config.interfaces, 'tf')
-        run: pip3 install tensorflow==$TF_VERSION
+        run: pip3 install tensorflow==$TF_VERSION keras==$TF_VERSION
 
       # Jax releases new version very frequently, so we should always build
       # to the latest release. We can always fix a version later if it breaks.