stateful dl smoke test (#1300) (#1330)

* stateful dl smoke test Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags: * remove version check Co-authored-by: Gokul <gokulg@meta.com>
pytorch · Sep 27, 2024 · 6b762ba · 6b762ba
1 parent ef62e00
commit 6b762ba
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
@@ -16,6 +16,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
@@ -26,3 +30,5 @@ def s3_test():
  options = parser.parse_args()
  if options.s3:
  s3_test()
+
+ stateful_dataloader_test()