From 6b762ba9cb42db8fde53f6f2b3adce990e88ef68 Mon Sep 17 00:00:00 2001 From: Andrew Ho Date: Fri, 27 Sep 2024 14:08:52 -0400 Subject: [PATCH] stateful dl smoke test (#1300) (#1330) * stateful dl smoke test Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags: * remove version check Co-authored-by: Gokul --- test/smoke_test/smoke_test.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index d3ad74ba2..6b4319843 100644 --- 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()