glenn-jocher pre-commit commited on
Commit
34b859a
1 Parent(s): 06bf8ef

Keras CI fix (#5486)

Browse files

* Keras CI fix

* pre-commit fixes

* Update ci-testing.yml

Co-authored-by: pre-commit <pre-commit@example.com>

.github/workflows/ci-testing.yml CHANGED
@@ -46,11 +46,12 @@ jobs:
46
  restore-keys: |
47
  ${{ runner.os }}-${{ matrix.python-version }}-pip-
48
 
 
49
  - name: Install dependencies
50
  run: |
51
  python -m pip install --upgrade pip
52
  pip install -qr requirements.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html
53
- pip install -q onnx tensorflow-cpu # for export
54
  python --version
55
  pip --version
56
  pip list
 
46
  restore-keys: |
47
  ${{ runner.os }}-${{ matrix.python-version }}-pip-
48
 
49
+ # Known Keras 2.7.0 issue: https://github.com/ultralytics/yolov5/pull/5486
50
  - name: Install dependencies
51
  run: |
52
  python -m pip install --upgrade pip
53
  pip install -qr requirements.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html
54
+ pip install -q onnx tensorflow-cpu keras==2.6.0 # for export
55
  python --version
56
  pip --version
57
  pip list
tutorial.ipynb CHANGED
@@ -1085,4 +1085,4 @@
1085
  "outputs": []
1086
  }
1087
  ]
1088
- }
 
1085
  "outputs": []
1086
  }
1087
  ]
1088
+ }