From e38a57039b318595d28671b0c9c9d6f3d97927c5 Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Tue, 11 Jul 2023 14:56:09 -0400 Subject: [PATCH] . --- tests/deepsparse/utils/onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/deepsparse/utils/onnx.py b/tests/deepsparse/utils/onnx.py index ffe6499845..34b95c9a47 100644 --- 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