Skip to content

Commit

Permalink
stateful dl smoke test (#1300) (#1330) (#1331)
Browse files Browse the repository at this point in the history
* stateful dl smoke test

Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

* remove version check

Co-authored-by: Gokul <[email protected]>
  • Loading branch information
andrewkho and gokulavasan authored Sep 27, 2024
1 parent 7af5272 commit c4f57b4
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions test/smoke_test/smoke_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -26,3 +30,5 @@ def s3_test():
options = parser.parse_args()
if options.s3:
s3_test()

stateful_dataloader_test()

0 comments on commit c4f57b4

Please sign in to comment.