Keras CI fix (#5486)

* Keras CI fix * pre-commit fixes * Update ci-testing.yml Co-authored-by: pre-commit <pre-commit@example.com>
ultralytics · Nov 3, 2021 · 34b859a · 34b859a
1 parent 06bf8ef
commit 34b859a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
@@ -46,11 +46,12 @@ jobs:
           restore-keys: |
             ${{ runner.os }}-${{ matrix.python-version }}-pip-
 
+      # Known Keras 2.7.0 issue: https://github.com/ultralytics/yolov5/pull/5486
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
           pip install -qr requirements.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html
-          pip install -q onnx tensorflow-cpu  # for export
+          pip install -q onnx tensorflow-cpu keras==2.6.0  # for export
           python --version
           pip --version
           pip list

diff --git a/tutorial.ipynb b/tutorial.ipynb
@@ -1085,4 +1085,4 @@
       "outputs": []
     }
   ]
-}
+}