test

huggingface · Oct 7, 2024 · a60d150 · a60d150
1 parent e5a65f2
commit a60d150
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/.github/workflows/test_bettertransformer.yml b/.github/workflows/test_bettertransformer.yml
@@ -31,11 +31,14 @@ jobs:
 
       - name: Install dependencies (stable)
         run: |
-          pip uninstall -y opencv-python
-          pip install opencv-python-headless
           pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
           pip install .[tests]
 
+      - name: Install headless opencv
+        run: |
+          pip uninstall -y opencv-python
+          pip install opencv-python-headless
+
       - name: Test with pytest (parallel)
         working-directory: tests
         run: |

diff --git a/.github/workflows/test_export_onnx_cli.yml b/.github/workflows/test_export_onnx_cli.yml
@@ -31,11 +31,14 @@ jobs:
 
       - name: Install dependencies
         run: |
-          pip uninstall -y opencv-python
-          pip install opencv-python-headless
           pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
           pip install .[tests,exporters] git+https://github.com/huggingface/accelerate.git
 
+      - name: Install headless opencv
+        run: |
+          pip uninstall -y opencv-python
+          pip install opencv-python-headless
+
       - name: Test with pytest (parallel)
         working-directory: tests
         run: |