Merge pull request #124 from DLHub-Argonne/fix_yaml

Fix tests for old versions of keras
DLHub-Argonne · Oct 22, 2021 · 9449f12 · 9449f12
2 parents 3e5c672 + 16e307a
commit 9449f12
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -65,7 +65,7 @@ jobs:
 
           # Adding keras and TF1 for some tests
           if [ -n "${{matrix.cfg.keras-version}}" ]; then
-            pip install keras==${{matrix.cfg.keras-version}} "tensorflow<2" "h5py<3"
+            pip install keras==${{matrix.cfg.keras-version}} "tensorflow<2" "h5py<3" "pyyaml<6"
           else
             # Otherwise, use TF2
             pip uninstall keras