From 9969d9717f634fba4f0be6aa4d771894eaff718f Mon Sep 17 00:00:00 2001 From: Elena Zhelezina Date: Wed, 28 Oct 2026 08:27:03 +0000 Subject: [PATCH] Arm backend: Support newer Model Explorer adapter import Signed-off-by: Elena Zhelezina Change-Id: Ib8766d737b77cddf79bb2ddbc7567662fed83282 --- backends/arm/vgf/quantization_visualization.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/backends/arm/vgf/quantization_visualization.py b/backends/arm/vgf/quantization_visualization.py index 9044b553f2a..bf47dae3cff 100644 --- a/backends/arm/vgf/quantization_visualization.py +++ b/backends/arm/vgf/quantization_visualization.py @@ -120,9 +120,17 @@ def _model_explorer_imports() -> tuple[Any, Any, Any, Any, Any]: from model_explorer.graph_builder import ( # type: ignore[import-not-found] KeyValue, ) - from model_explorer.pytorch_exported_program_adater_impl import ( # type: ignore - PytorchExportedProgramAdapterImpl, - ) + + try: + # ai-edge-model-explorer >= 0.1.34 corrected this module name. + from model_explorer.pytorch_exported_program_adapter_impl import ( # type: ignore + PytorchExportedProgramAdapterImpl, + ) + except ImportError: + # ai-edge-model-explorer <= 0.1.33 used the 'adater' spelling. + from model_explorer.pytorch_exported_program_adater_impl import ( # type: ignore + PytorchExportedProgramAdapterImpl, + ) except ImportError as exc: raise ImportError( "Model Explorer is required for VGF quantization visualization. "