Merge branch 'main' into andrewkh/cleanup-datapipes-dlv2

pytorch · Sep 27, 2024 · 4b13f2f · 4b13f2f
2 parents 17ca3e2 + 6b762ba
commit 4b13f2f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -15,6 +15,10 @@ def s3_test():
     from torchdata._torchdata import S3Handler
 
 
+def stateful_dataloader_test():
+    from torchdata.stateful_dataloader import StatefulDataLoader
+
+
 if __name__ == "__main__":
     r"""
     TorchData Smoke Test
@@ -25,3 +29,5 @@ def s3_test():
     options = parser.parse_args()
     if options.s3:
         s3_test()
+
+    stateful_dataloader_test()