Skip to content

Commit d93861d

Browse files
committed
NXP backend: Implemented the loader for eIQ Neutron SDK converter.
1 parent cf28248 commit d93861d

File tree

2 files changed

+13
-5
lines changed

2 files changed

+13
-5
lines changed

backends/nxp/backend/neutron_converter_manager.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,19 +35,27 @@ def __init__(
3535
module.name
3636
for module in pkgutil.iter_modules()
3737
if module.name.startswith("neutron_converter")
38+
or module.name == "eiq_neutron_sdk"
3839
]
3940

40-
requested_module_name = f"neutron_converter_{neutron_converter_flavor}"
41+
if neutron_converter_flavor:
42+
requested_module_name = f"neutron_converter_{neutron_converter_flavor}"
43+
logging.warning(
44+
"The use of converter flavors will be deprecated. Use empty string to select 'eiq_neutron_sdk' module."
45+
)
46+
else:
47+
requested_module_name = "eiq_neutron_sdk"
48+
4149
if requested_module_name not in neutron_converter_modules:
4250
if len(neutron_converter_modules) > 0:
4351
raise RuntimeError(
44-
f"Neutron Converter module with flavor '{neutron_converter_flavor}' "
52+
f"Neutron Converter module '{requested_module_name}' "
4553
f"not found. Available modules: {neutron_converter_modules}."
4654
)
4755
else:
4856
raise RuntimeError(
49-
f"Neutron Converter module with flavor '{neutron_converter_flavor}' "
50-
f"not found. Install 'neutron_converter_[flavor]' Python package."
57+
f"Neutron Converter module '{requested_module_name}' "
58+
f"not found. Install 'eiq_neutron_sdk' or 'neutron_converter_[flavor]' Python package."
5159
)
5260

5361
self.neutron_converter = importlib.import_module(

backends/nxp/tests/test_neutron_converter_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,6 @@ def test__conv2d_neutron_conversion__invalid_flavor():
5454
with pytest.raises(RuntimeError) as excinfo:
5555
_ = NeutronConverterManager("bad_flavor").convert(tflite_model, "imxrt700")
5656

57-
assert "Neutron Converter module with flavor 'bad_flavor' not found." in str(
57+
assert "Neutron Converter module 'neutron_converter_bad_flavor' not found." in str(
5858
excinfo
5959
)

0 commit comments

Comments
 (0)