Skip to content

Commit f84be56

Browse files
committed
fix: distingush engines based on compilation settings in addition to graph structure
Signed-off-by: Naren Dasan <[email protected]>
1 parent 21299ae commit f84be56

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/py/dynamo/models/test_models_export.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
# type: ignore
22
import importlib
33
import unittest
4+
from importlib import metadata
45

56
import pytest
67
import timm
@@ -250,7 +251,7 @@ def calibrate_loop(model):
250251

251252
@unittest.skipIf(
252253
not importlib.util.find_spec("modelopt")
253-
or Version(importlib.metadata.version("modelopt")) < Version("0.16.1"),
254+
or Version(metadata.version("nvidia-modelopt")) < Version("0.16.1"),
254255
"modelopt 0.16.1 or later is required Int8 quantization is supported in modelopt since 0.16.1 or later",
255256
)
256257
@pytest.mark.unit

0 commit comments

Comments
 (0)