From 63d7f5cf1efdb9de98e42a8dcc91e04a76c36056 Mon Sep 17 00:00:00 2001
From: Reuven <44209964+reuvenperetz@users.noreply.github.com>
Date: Mon, 18 Dec 2023 12:40:12 +0200
Subject: [PATCH] Update compatibility_torch_tests.yml

---
 .github/workflows/compatibility_torch_tests.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/compatibility_torch_tests.yml b/.github/workflows/compatibility_torch_tests.yml
index 1dec2c9..4473636 100644
--- 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  }}
\ No newline at end of file
+          python tests/compatibility_tests/torch_comp_tests/compatibility_activation_load_model_test_suite.py ${{  inputs.save_version  }}