print version that was found

pytorch · Jul 26, 2024 · 0a1fc12 · 0a1fc12
1 parent 322712c
commit 0a1fc12
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
@@ -32,5 +32,5 @@ def stateful_dataloader_test():
         s3_test()
 
     if torchdata.__version__ != "0.8.0":
-        raise Exception("TorchData version is not 0.8.0")
+        raise Exception(f"TorchData version is not 0.8.0, found {torchdata.__version__}")
     stateful_dataloader_test()