Lluo/fix merge issue (#3162)

pytorch · Sep 15, 2024 · 6717ddc · 6717ddc
1 parent a662599
commit 6717ddc
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/py/dynamo/models/test_models_export.py b/tests/py/dynamo/models/test_models_export.py
@@ -1,4 +1,5 @@
 # type: ignore
+import importlib
 import platform
 import unittest
 from importlib import metadata
@@ -250,8 +251,10 @@ def calibrate_loop(model):
 
 
 @unittest.skipIf(
-    modelopt.__version__ < "0.16.1",
-    "Int8 quantization is supported in modelopt since 0.16.1 or later",
+    not importlib.util.find_spec("modelopt")
+    or platform.system() != "Linux"
+    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
 def test_base_int8(ir):