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 e2549cc commit aaf10b2Copy full SHA for aaf10b2
tests/onnx/quantization/test_weights_compression.py
@@ -18,6 +18,7 @@
18
from onnx import helper
19
from onnx import numpy_helper
20
from onnxruntime import InferenceSession
21
+from packaging import version
22
23
from nncf import CompressWeightsMode
24
from nncf.onnx.graph.onnx_helper import get_edge_shape
@@ -269,7 +270,7 @@ def test_compression_with_inference(mode):
269
270
271
def test_matmulnbits():
272
rtol = 1e-5
- if onnxruntime.__version__ < "1.21.1":
273
+ if version.parse(onnxruntime.__version__) < version.parse("1.21.1"):
274
rtol = 1e-3
275
276
np.random.seed(42)
0 commit comments