combine workflows

huggingface · Feb 28, 2024 · 28044ac · 28044ac
1 parent fd696a6
commit 28044ac
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/.github/workflows/test_model.yml b/.github/workflows/test_model.yml
@@ -71,6 +71,16 @@ jobs:
 #        run: |
 #          python -c 'import os; import sys; SRC_DIRS = [os.path.join("/usr/local/lib/python3.10/dist-packages/transformers/examples/pytorch", dirname) for dirname in ["image-classification"]]; sys.path.extend(SRC_DIRS); import run_image_classification'
 
+      - name: Echo 5
+        shell: bash
+        run: |
+          git clone https://github.com/huggingface/transformers.git
+
+      - name: Echo 6
+        shell: bash
+        run: |
+          ls -la ./transformers/examples/pytorch
+
       - name: run tests
         run: |
           python3 -m pytest -v tests/models
diff --git a/tests/models/vit/test_training_vit.py b/tests/models/vit/test_training_vit.py
@@ -5,7 +5,7 @@
 
 
 # TODO: Find a way to get the installed `transformers` directory.
-SRC_DIRS = [os.path.join("/usr/local/lib/python3.10/dist-packages/transformers/examples/pytorch", dirname) for dirname in ["image-classification"]]
+SRC_DIRS = [os.path.join("/__w/Google-Cloud-Containers/Google-Cloud-Containers/transformers/examples/pytorch", dirname) for dirname in ["image-classification"]]
 sys.path.extend(SRC_DIRS)