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