We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 21299ae commit f84be56Copy full SHA for f84be56
tests/py/dynamo/models/test_models_export.py
@@ -1,6 +1,7 @@
1
# type: ignore
2
import importlib
3
import unittest
4
+from importlib import metadata
5
6
import pytest
7
import timm
@@ -250,7 +251,7 @@ def calibrate_loop(model):
250
251
252
@unittest.skipIf(
253
not importlib.util.find_spec("modelopt")
- or Version(importlib.metadata.version("modelopt")) < Version("0.16.1"),
254
+ or Version(metadata.version("nvidia-modelopt")) < Version("0.16.1"),
255
"modelopt 0.16.1 or later is required Int8 quantization is supported in modelopt since 0.16.1 or later",
256
)
257
@pytest.mark.unit
0 commit comments