diff options
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 |