From e223fb478b0c43272d9e304e6a0b98d25ad2ba1f Mon Sep 17 00:00:00 2001 From: Seokjun An Date: Wed, 15 Jan 2025 20:28:52 +0900 Subject: [PATCH] Fix unittest failure Signed-off-by: Seokjun An --- .../src/python/aimet_torch/v2/experimental/onnx/_export.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/TrainingExtensions/torch/src/python/aimet_torch/v2/experimental/onnx/_export.py b/TrainingExtensions/torch/src/python/aimet_torch/v2/experimental/onnx/_export.py index 3dfc2fbc770..632667dcc14 100644 --- a/TrainingExtensions/torch/src/python/aimet_torch/v2/experimental/onnx/_export.py +++ b/TrainingExtensions/torch/src/python/aimet_torch/v2/experimental/onnx/_export.py @@ -46,7 +46,6 @@ from torch.onnx import is_in_onnx_export, symbolic_helper from aimet_torch.v2.utils import patch_attr -from aimet_torch.quantization.base import QuantizerBase aimet_opset = onnxscript.values.Opset(domain="aimet", version=1) @@ -219,6 +218,8 @@ def export(model: torch.nn.Module, *args, **kwargs): @contextmanager def _precompute_encodings(model: torch.nn.Module): + # pylint: disable=import-outside-toplevel + from aimet_torch.quantization.base import QuantizerBase with ExitStack() as stack: for q in model.modules(): if isinstance(q, QuantizerBase):