diff --git a/tests/py/dynamo/models/test_models_export.py b/tests/py/dynamo/models/test_models_export.py index d547a2e0bf..69944aa746 100644 --- 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):