starts with instead of strict equality

pytorch · Jul 26, 2024 · 38beea5 · 38beea5
1 parent 0a1fc12
commit 38beea5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -31,6 +31,6 @@ def stateful_dataloader_test():
     if options.s3:
         s3_test()
 
-    if torchdata.__version__ != "0.8.0":
+    if torchdata.__version__.startswith("0.8.0"):
         raise Exception(f"TorchData version is not 0.8.0, found {torchdata.__version__}")
     stateful_dataloader_test()