Skip to content

Commit aaf10b2

Browse files
Follow up # 3525 (#3530)
### Changes Follow up for #3525
1 parent e2549cc commit aaf10b2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/onnx/quantization/test_weights_compression.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
from onnx import helper
1919
from onnx import numpy_helper
2020
from onnxruntime import InferenceSession
21+
from packaging import version
2122

2223
from nncf import CompressWeightsMode
2324
from nncf.onnx.graph.onnx_helper import get_edge_shape
@@ -269,7 +270,7 @@ def test_compression_with_inference(mode):
269270

270271
def test_matmulnbits():
271272
rtol = 1e-5
272-
if onnxruntime.__version__ < "1.21.1":
273+
if version.parse(onnxruntime.__version__) < version.parse("1.21.1"):
273274
rtol = 1e-3
274275

275276
np.random.seed(42)

0 commit comments

Comments
 (0)