try test_state_dict only

pytorch · May 9, 2024 · ddbfbe6 · ddbfbe6
1 parent 9f59102
commit ddbfbe6
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/test_dataloader.py
+        run: pytest --durations=0 --no-header -v test/stateful_dataloader/test_state_dict.py