Limit TF export to tensorflow<2.15.1 (#12800)

* Update ci-testing.yml Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * bump tests to 3.12 Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update ci-testing.yml Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update ci-testing.yml Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update ci-testing.yml Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> --------- Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
ultralytics · Mar 10, 2024 · a083914 · a083914
1 parent b2ffe05
commit a083914
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
@@ -18,7 +18,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest]
-        python-version: ["3.11"] # requires python<=3.10
+        python-version: ["3.11"] # requires python<=3.11
         model: [yolov5n]
     steps:
       - uses: actions/checkout@v4
@@ -29,7 +29,7 @@ jobs:
       - name: Install requirements
         run: |
           python -m pip install --upgrade pip wheel
-          pip install -r requirements.txt coremltools openvino-dev tensorflow-cpu --extra-index-url https://download.pytorch.org/whl/cpu
+          pip install -r requirements.txt coremltools openvino-dev "tensorflow-cpu<2.15.1" --extra-index-url https://download.pytorch.org/whl/cpu
           yolo checks
           pip list
       - name: Benchmark DetectionModel
@@ -51,16 +51,10 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [ubuntu-latest, windows-latest] # macos-latest bug https://github.com/ultralytics/yolov5/pull/9049
+        os: [ubuntu-latest, windows-latest, macos-14] # macos-latest bug https://github.com/ultralytics/yolov5/pull/9049
         python-version: ["3.11"]
         model: [yolov5n]
         include:
-          - os: ubuntu-latest
-            python-version: "3.8" # '3.6.8' min
-            model: yolov5n
-          - os: ubuntu-latest
-            python-version: "3.9"
-            model: yolov5n
           - os: ubuntu-latest
             python-version: "3.8" # torch 1.8.0 requires python >=3.6, <=3.8
             model: yolov5n