From 0a1fc12fe2f260605445c9deceeb838330261b28 Mon Sep 17 00:00:00 2001 From: Gokul Gunasekaran Date: Fri, 26 Jul 2024 12:27:40 -0700 Subject: [PATCH] print version that was found --- test/smoke_test/smoke_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index d3bf55373..3531d8478 100644 --- 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()