support for new step

nod-ai · Mar 14, 2024 · 6904a45 · 6904a45
1 parent 84c7b4d
commit 6904a45
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/test_e2eshark.yml b/.github/workflows/test_e2eshark.yml
@@ -132,7 +132,7 @@ jobs:
           path: turbine
 
   e2eshark:
-    needs: [turbine]
+    needs: [turbine, torch-mlir]
     strategy:
       matrix:
         version: [3.11]
@@ -170,7 +170,7 @@ jobs:
 
       - name: Run Tests
         run: |
+          source ${E2E_VENV_DIR}/bin/activate
           cd e2eshark
           python ./run.py -c ../../torch-mlir/build --report --cachedir ~/.cache/huggingface --mode turbine --tests pytorch/models/beit-base-patch16-224-pt22k-ft22k pytorch/models/bge-base-en-v1.5 pytorch/models/mit-b0 pytorch/models/bert-large-uncased pytorch/models/deit-small-distilled-patch16-224 --cleanup
-          deactivate
         working-directory: ./test-suite