stateful dl smoke test

Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

Author: gokulavasan

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,7 @@ def s3_test():
     options = parser.parse_args()
     if options.s3:
         s3_test()
+
+    if torchdata.__version__ != "0.8.0":
+        raise Exception("TorchData version is not 0.8.0")
+    stateful_dataloader_test()