CI test

hpcaitech · Aug 29, 2023 · 7b90056 · 7b90056
1 parent 0e3a5c1
commit 7b90056
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/build_on_pr.yml b/.github/workflows/build_on_pr.yml
@@ -208,7 +208,7 @@ jobs:
 
       - name: Execute Unit Testing
         run: |
-          CURL_CA_BUNDLE="" PYTHONPATH=$PWD pytest -m "not largedist" --testmon --testmon-cov=. --durations=10 tests/
+          CURL_CA_BUNDLE="" PYTHONPATH=$PWD pytest -m "not largedist" -s --testmon --testmon-cov=. --durations=10 tests/test_lazy/test_models.py
         env:
           DATA: /data/scratch/cifar-10
           NCCL_SHM_DISABLE: 1

diff --git a/tests/test_lazy/test_models.py b/tests/test_lazy/test_models.py
@@ -10,6 +10,7 @@
 def test_torchvision_models_lazy_init(subset, default_device):
     sub_model_zoo = model_zoo.get_sub_registry(subset)
     for name, entry in sub_model_zoo.items():
+        print(subset, default_device, name)
         # TODO(ver217): lazy init does not support weight norm, skip these models
         if name in ('torchaudio_wav2vec2_base',
                     'torchaudio_hubert_base') or name.startswith('transformers_llama') or name.startswith(