test

pytorch · Sep 15, 2024 · f618694 · f618694
1 parent 33ae883
commit f618694
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tests/py/dynamo/models/test_models_export.py b/tests/py/dynamo/models/test_models_export.py
@@ -1,5 +1,5 @@
 # type: ignore
-import importlib
+import platform
 import unittest
 from importlib import metadata
 
@@ -250,10 +250,8 @@ def calibrate_loop(model):
 
 
 @unittest.skipIf(
-    not importlib.util.find_spec("modelopt")
-    or platform.system() != "Linux"
-    or Version(metadata.version("nvidia-modelopt")) < Version("0.17.0"),
-    "modelopt 0.17.0 or later is required, Int8 quantization is supported in modelopt since 0.17.0 or later for linux",
+    modelopt.__version__ < "0.16.1",
+    "Int8 quantization is supported in modelopt since 0.16.1 or later",
 )
 @pytest.mark.unit
 def test_base_int8(ir):