Update compatibility_torch_tests.yml

ofirgo · Dec 18, 2023 · 63d7f5c · 63d7f5c
1 parent 1525b02
commit 63d7f5c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/compatibility_torch_tests.yml b/.github/workflows/compatibility_torch_tests.yml
@@ -33,7 +33,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements.txt
-          pip install torch==${{  inputs.torch_version  }}
+          pip install torch==${{  inputs.torch_version  }} onnx onnxruntime onnxruntime-extensions
       - name: Checkout to MCT Quantizers requested tag for saving test models
         run: |
           git checkout tags/${{  inputs.save_version  }}
@@ -57,4 +57,4 @@ jobs:
           echo "Updated PYTHONPATH: $PYTHONPATH"
           cd ..
           python tests/compatibility_tests/torch_comp_tests/compatibility_weights_load_model_test_suite.py ${{  inputs.save_version  }}
-          python tests/compatibility_tests/torch_comp_tests/compatibility_activation_load_model_test_suite.py ${{  inputs.save_version  }}
+          python tests/compatibility_tests/torch_comp_tests/compatibility_activation_load_model_test_suite.py ${{  inputs.save_version  }}