From 6ec6beb485ae29fcd9dfbe9607bf8b252d0f2c28 Mon Sep 17 00:00:00 2001 From: Gokul Date: Fri, 27 Sep 2024 08:36:33 -0700 Subject: [PATCH] stateful dl smoke test (#1300) * stateful dl smoke test Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags: * remove version check --- 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()