File tree Expand file tree Collapse file tree 2 files changed +13
-5
lines changed
Expand file tree Collapse file tree 2 files changed +13
-5
lines changed Original file line number Diff line number Diff 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+ print (
44+ "Warning: 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 (
Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments