Keras CI fix

ultralytics · Nov 3, 2021 · 6fd5cfd · 6fd5cfd
1 parent 06bf8ef
commit 6fd5cfd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
@@ -50,7 +50,7 @@ jobs:
         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