.

neuralmagic · Jul 11, 2023 · e38a570 · e38a570
1 parent 8d00f18
commit e38a570
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/deepsparse/utils/onnx.py b/tests/deepsparse/utils/onnx.py
@@ -16,7 +16,7 @@
 import onnx
 
 import pytest
-from deepsparse.utils import override_onnx_batch_size, override_onnx_input_shapes
+from deepsparse.utils.onnx import override_onnx_batch_size, override_onnx_input_shapes
 from sparsezoo import Model