Improve INC CI test torch version (#1027)

* improve inc and ipex test ci torch version Signed-off-by: changwa1 <[email protected]> * only improve test_inc.yml Signed-off-by: changwa1 <[email protected]> * recover test_ipex.yml Signed-off-by: changwa1 <[email protected]> --------- Signed-off-by: changwa1 <[email protected]>
huggingface · Nov 25, 2024 · 8fcc0ec · 8fcc0ec
1 parent a9c4772
commit 8fcc0ec
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/.github/workflows/test_inc.yml b/.github/workflows/test_inc.yml
@@ -18,7 +18,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        torch-version: ["2.2.0", "2.3.*", "2.4.*"]
+        torch-version: ["2.4.*", "2.5.0"]
 
     runs-on: ubuntu-22.04
 
@@ -37,10 +37,6 @@ jobs:
           pip install torch==${{ matrix.torch-version }} torchaudio torchvision --index-url https://download.pytorch.org/whl/cpu
           pip install .[neural-compressor,ipex,diffusers,peft,tests] transformers[testing] intel-extension-for-pytorch==${{ matrix.torch-version }}
 
-      - if: ${{ matrix.torch-version == '2.2.0' }}
-        name: Downgrade Numpy
-        run: pip install numpy==1.*
-
       - name: Assert versions
         run: |
           python -c "import torch; print(torch.__version__); assert torch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"

diff --git a/.github/workflows/test_ipex.yml b/.github/workflows/test_ipex.yml
@@ -50,4 +50,4 @@ jobs:
 
       - name: Test with Pytest
         run: |
-          pytest tests/ipex
+          pytest tests/ipex