diff options
author | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2023-08-24 16:38:47 +0100 |
---|---|---|
committer | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2023-09-05 14:20:08 +0100 |
commit | e5a0bc3ecd4d9c46ead3b8217584eaa916a3afa4 (patch) | |
tree | 94c348fcef50326a755a049a2a4027f588211f8b /src/mlia/target/ethos_u | |
parent | 900c3c52b681e0b8a4454e2e2cf29265d53a2c98 (diff) | |
download | mlia-e5a0bc3ecd4d9c46ead3b8217584eaa916a3afa4.tar.gz |
MLIA-961 Update tox dependencies
- Update version dependencies in the tox.ini
- Fix linter issues
Change-Id: I04c3a841ee2646a865dab037701d66c28792f2a4
Signed-off-by: Benjamin Klimczak <benjamin.klimczak@arm.com>
Diffstat (limited to 'src/mlia/target/ethos_u')
-rw-r--r-- | src/mlia/target/ethos_u/advisor.py | 6 | ||||
-rw-r--r-- | src/mlia/target/ethos_u/data_collection.py | 2 | ||||
-rw-r--r-- | src/mlia/target/ethos_u/reporters.py | 2 |
3 files changed, 5 insertions, 5 deletions
diff --git a/src/mlia/target/ethos_u/advisor.py b/src/mlia/target/ethos_u/advisor.py index 714d6a4..d2c308a 100644 --- a/src/mlia/target/ethos_u/advisor.py +++ b/src/mlia/target/ethos_u/advisor.py @@ -54,7 +54,7 @@ class EthosUInferenceAdvisor(DefaultInferenceAdvisor): if is_tflite_model(model): # TensorFlow Lite models do not support optimization (only performance)! if context.category_enabled(AdviceCategory.OPTIMIZATION): - raise Exception( + raise RuntimeError( "Optimizations are not supported for TensorFlow Lite files." ) if context.category_enabled(AdviceCategory.PERFORMANCE): @@ -170,7 +170,7 @@ def _get_config_parameters( backends = extra_args.get("backends") if backends is not None: if not is_list_of(backends, str): - raise Exception("Backends value has wrong format") + raise ValueError("Backends value has wrong format.") advisor_parameters["ethos_u_inference_advisor"]["backends"] = backends @@ -179,7 +179,7 @@ def _get_config_parameters( optimization_targets = _DEFAULT_OPTIMIZATION_TARGETS if not is_list_of(optimization_targets, dict): - raise Exception("Optimization targets value has wrong format") + raise ValueError("Optimization targets value has wrong format.") advisor_parameters.update( { diff --git a/src/mlia/target/ethos_u/data_collection.py b/src/mlia/target/ethos_u/data_collection.py index 8348393..0654143 100644 --- a/src/mlia/target/ethos_u/data_collection.py +++ b/src/mlia/target/ethos_u/data_collection.py @@ -178,7 +178,7 @@ class EthosUOptimizationPerformance(ContextAwareDataCollector): ) -> list[list[OptimizationSettings]]: """Parse optimization parameters.""" if not is_list_of(optimizations, list): - raise Exception("Optimization parameters expected to be a list") + raise ValueError("Optimization parameters expected to be a list.") return [ [ diff --git a/src/mlia/target/ethos_u/reporters.py b/src/mlia/target/ethos_u/reporters.py index 00e68b5..711f036 100644 --- a/src/mlia/target/ethos_u/reporters.py +++ b/src/mlia/target/ethos_u/reporters.py @@ -390,6 +390,6 @@ def ethos_u_formatters(data: Any) -> Callable[[Any], Report]: report = report_tflite_compatiblity else: - raise Exception(f"Unable to find appropriate formatter for {data}") + raise RuntimeError(f"Unable to find appropriate formatter for {data}.") return report |