test

pytorch · Sep 15, 2024 · 310a675 · 310a675
1 parent dd2325d
commit 310a675
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/py/dynamo/models/test_models_export.py b/tests/py/dynamo/models/test_models_export.py
@@ -253,7 +253,7 @@ def calibrate_loop(model):
 @unittest.skipIf(
     not importlib.util.find_spec("modelopt")
     or platform.system() != "Linux"
-    or modelopt.__version__ < "0.17.0",
+    or Version(metadata.version("nvidia-modelopt")) < Version("0.17.0"),
     "Int8 quantization is supported in modelopt since 0.17.0 or later for Linux",
 )
 @pytest.mark.unit