From 9cfa10489915c291e6cced8ec754f5a78ece6d40 Mon Sep 17 00:00:00 2001 From: Michael Tuttle Date: Wed, 20 Nov 2024 00:28:33 -0800 Subject: [PATCH] Fix pylint failures from cyclic import (#3514) Signed-off-by: Michael Tuttle --- TrainingExtensions/torch/src/python/aimet_torch/v2/nn/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TrainingExtensions/torch/src/python/aimet_torch/v2/nn/base.py b/TrainingExtensions/torch/src/python/aimet_torch/v2/nn/base.py index 8046c482866..f13071b17bf 100644 --- a/TrainingExtensions/torch/src/python/aimet_torch/v2/nn/base.py +++ b/TrainingExtensions/torch/src/python/aimet_torch/v2/nn/base.py @@ -225,7 +225,7 @@ def wrapper(quantized_cls): # TODO: This unfortunately relies on the **class name** of the module, not the real type # of the module due to the limitation of v1 implementation. # Should redefine `aimet_to_to_backend_op_name_map` as `Dict[Type[Module], str]` - from aimet_torch.translation_mapping import aimet_op_to_backend_op_name_map + from aimet_torch.translation_mapping import aimet_op_to_backend_op_name_map # pylint:disable = cyclic-import backend_op_name = aimet_op_to_backend_op_name_map.get(module_cls, None) if backend_op_name: aimet_op_to_backend_op_name_map[quantized_cls] = backend_op_name