fix hf token command in unit_test_model_trainer_gpu.yaml

fix pytest command execution directory
KevKibe · Nov 15, 2024 · f657e77 · f657e77
1 parent 5d31739
commit f657e77
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/.github/workflows/training_tests.yaml → .../workflows/unit_test_data_processing.yaml b/.github/workflows/training_tests.yaml → .../workflows/unit_test_data_processing.yaml
@@ -44,4 +44,4 @@ jobs:
         env:
           HF_TOKEN: ${{ secrets.HF_TOKEN }}
           WANDB_TOKEN: ${{ secrets.WANDB_TOKEN }}
-        run: pytest -vv tests/test_audio_processor.py src/tests/test_data_prep.py src/tests/test_load_dataset.py
+        run: pytest -vv tests/test_audio_processor.py tests/test_data_prep.py tests/test_load_dataset.py
diff --git a/.github/workflows/unit_test_deployment.yaml b/.github/workflows/unit_test_deployment.yaml
@@ -44,4 +44,4 @@ jobs:
         env:
           HF_TOKEN: ${{ secrets.HF_TOKEN }}
           WANDB_TOKEN: ${{ secrets.WANDB_TOKEN }}
-        run: pytest -vv tests/test_model_optimization.py src/tests/test_transcription_pipeline.py
+        run: pytest -vv tests/test_model_optimization.py tests/test_transcription_pipeline.py
diff --git a/.github/workflows/unit_test_model_trainer_gpu.yaml b/.github/workflows/unit_test_model_trainer_gpu.yaml
@@ -11,7 +11,7 @@ jobs:
         uses: actions/checkout@v3
 
       - name: Execute Tests with GPU Support
-        uses: KevKibe/kaggle-script-action@v1.0.1
+        uses: KevKibe/kaggle-script-action@v1.0.3
         env:
           HF_TOKEN: ${{ secrets.HF_TOKEN }}
           WANDB_TOKEN: ${{ secrets.WANDB_TOKEN }}
@@ -20,8 +20,7 @@ jobs:
           key: ${{ secrets.KAGGLE_KEY }}
           title: "Test PEFT Finetuning"
           custom_script: |
-            huggingface-cli login --token ${{ secrets.HF_TOKEN }}
-            pytest -vv tests/test_model_trainer_gpu.py
+            huggingface-cli login --token ${{ secrets.HF_TOKEN }} && pytest -vv tests/test_model_trainer_gpu.py
           enable_internet: true
           enable_gpu: true
           enable_tpu: false