Address comment

Author: gokulavasan

test/smoke_test/smoke_test.py

@@ -30,6 +30,7 @@ def stateful_dataloader_test():
     options = parser.parse_args()
     if options.s3:
         s3_test()
-    
-    if torchdata.__version__ >= "0.8.0": 
-        stateful_dataloader_test()
+
+    if torchdata.__version__ != "0.8.0":
+        raise Exception("TorchData version is not 0.8.0")
+    stateful_dataloader_test()