check if test_state_dict or test_dataloader is the culprit

pytorch · May 9, 2024 · 0264ee9 · 0264ee9
1 parent 11e16da
commit 0264ee9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/stateful_dataloader_ci.yml b/.github/workflows/stateful_dataloader_ci.yml
@@ -83,4 +83,4 @@ jobs:
           cd ..
       - name: Run StatefulDataLoader tests with pytest
         if: ${{ ! contains(github.event.pull_request.labels.*.name, 'ciflow/slow') }}
-        run: pytest --durations=0 --no-header -v test/stateful_dataloader
+        run: pytest --durations=0 --no-header -v test/stateful_dataloader/test_dataloader.py