diff options
Diffstat (limited to 'src/mlia/target/tosa')
-rw-r--r-- | src/mlia/target/tosa/__init__.py | 1 | ||||
-rw-r--r-- | src/mlia/target/tosa/reporters.py | 4 |
2 files changed, 3 insertions, 2 deletions
diff --git a/src/mlia/target/tosa/__init__.py b/src/mlia/target/tosa/__init__.py index 3830ce5..12077bd 100644 --- a/src/mlia/target/tosa/__init__.py +++ b/src/mlia/target/tosa/__init__.py @@ -14,4 +14,5 @@ registry.register( advisor_factory_func=configure_and_get_tosa_advisor, target_profile_cls=TOSAConfiguration, ), + pretty_name="TOSA", ) diff --git a/src/mlia/target/tosa/reporters.py b/src/mlia/target/tosa/reporters.py index 5c015ff..9575978 100644 --- a/src/mlia/target/tosa/reporters.py +++ b/src/mlia/target/tosa/reporters.py @@ -42,13 +42,13 @@ class MetadataDisplay: # pylint: disable=too-few-public-methods self.model_name = model_meta.model_name -def report_target(target: TOSAConfiguration) -> Report: +def report_target(target_config: TOSAConfiguration) -> Report: """Generate report for the target.""" return NestedReport( "Target information", "target", [ - ReportItem("Target", alias="target", value=target.target), + ReportItem("Target", alias="target", value=target_config.target), ], ) |