test

huggingface · Oct 27, 2024 · 60580a1 · 60580a1
1 parent 4305232
commit 60580a1
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 9 deletions.
diff --git a/.github/workflows/quality.yml b/.github/workflows/quality.yml
@@ -1,4 +1,4 @@
-name: Code Quality Checks
+name: Quality Checks
 on:
   push:
     branches:

diff --git a/.github/workflows/security.yml b/.github/workflows/security.yml
@@ -1,4 +1,4 @@
-name: Security Check
+name: Security Checks
 
 on:
   push:

diff --git a/.github/workflows/test_generation.yml b/.github/workflows/test_generation.yml
@@ -18,7 +18,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: [3.9]
+        python-version: ["3.9"]
         os: [ubuntu-latest]
 
     runs-on: ${{ matrix.os }}

diff --git a/.github/workflows/test_inc.yml b/.github/workflows/test_inc.yml
@@ -18,8 +18,8 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        torch-version: ["2.3.*", "2.4.*"]
-        python-version: [3.9]
+        torch-version: ["2.2.0", "2.4.*"]
+        python-version: ["3.9"]
         os: [ubuntu-latest]
 
     runs-on: ${{ matrix.os }}
@@ -40,6 +40,11 @@ jobs:
           pip install intel-extension-for-pytorch==${{ matrix.torch-version }}
           pip install .[neural-compressor,ipex,diffusers,peft,tests]
 
+      - name: Assert versions
+        run: |
+          python -c "import torch; print(torch.__version__); assert torch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"
+          python -c "import intel_extension_for_pytorch; print(intel_extension_for_pytorch.__version__); assert intel_extension_for_pytorch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"
+
       - name: Test with Pytest
         run: |
           pytest tests/neural_compressor/ -s -vvvv --durations=0
diff --git a/.github/workflows/test_ipex.yml b/.github/workflows/test_ipex.yml
@@ -44,9 +44,9 @@ jobs:
 
       - name: Assert versions
         run: |
-          python -c "import torch; assert torch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"
-          python -c "import transformers; assert transformers.__version__.startswith('${{ matrix.transformers-version }}'.replace('.*', ''))"
-          python -c "import intel_extension_for_pytorch; assert intel_extension_for_pytorch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"
+          python -c "import torch; print(torch.__version__); assert torch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"
+          python -c "import intel_extension_for_pytorch; print(intel_extension_for_pytorch.__version__); assert intel_extension_for_pytorch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"
+          python -c "import transformers; print(transformers.__version__); assert transformers.__version__.startswith('${{ matrix.transformers-version }}'.replace('.*', ''))"
 
       - if: ${{ matrix.torch-version == '2.2.*' }}
         name: Downgrade numpy for torch 2.2.*

diff --git a/.github/workflows/test_openvino.yml b/.github/workflows/test_openvino.yml
@@ -49,7 +49,7 @@ jobs:
 
       - name: Assert versions
         run: |
-          python -c "import transformers; assert transformers.__version__.startswith('${{ matrix.transformers-version }}'.replace('.*', ''))"
+          python -c "import transformers; print(transformers.__version__); assert transformers.__version__.startswith('${{ matrix.transformers-version }}'.replace('.*', ''))"
 
       - if: ${{ matrix.transformers-version == '4.36.0' }}
         run: pip install accelerate==0.*