diff options
Diffstat (limited to 'src/mlia/target/tosa')
-rw-r--r-- | src/mlia/target/tosa/events.py | 2 | ||||
-rw-r--r-- | src/mlia/target/tosa/handlers.py | 2 | ||||
-rw-r--r-- | src/mlia/target/tosa/reporters.py | 12 |
3 files changed, 8 insertions, 8 deletions
diff --git a/src/mlia/target/tosa/events.py b/src/mlia/target/tosa/events.py index cbfd199..06a05af 100644 --- a/src/mlia/target/tosa/events.py +++ b/src/mlia/target/tosa/events.py @@ -17,7 +17,7 @@ class TOSAAdvisorStartedEvent(Event): """Event with TOSA advisor parameters.""" model: Path - device: TOSAConfiguration + target: TOSAConfiguration tosa_metadata: MetadataDisplay | None diff --git a/src/mlia/target/tosa/handlers.py b/src/mlia/target/tosa/handlers.py index f222823..131afa7 100644 --- a/src/mlia/target/tosa/handlers.py +++ b/src/mlia/target/tosa/handlers.py @@ -25,7 +25,7 @@ class TOSAEventHandler(WorkflowEventsHandler, TOSAAdvisorEventHandler): def on_tosa_advisor_started(self, event: TOSAAdvisorStartedEvent) -> None: """Handle TOSAAdvisorStartedEvent event.""" - self.reporter.submit(event.device) + self.reporter.submit(event.target) self.reporter.submit(event.tosa_metadata) def on_collected_data(self, event: CollectedDataEvent) -> None: diff --git a/src/mlia/target/tosa/reporters.py b/src/mlia/target/tosa/reporters.py index 283f61f..5c015ff 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_device(device: TOSAConfiguration) -> Report: - """Generate report for the device.""" +def report_target(target: TOSAConfiguration) -> Report: + """Generate report for the target.""" return NestedReport( - "Device information", - "device", + "Target information", + "target", [ - ReportItem("Target", alias="target", value=device.target), + ReportItem("Target", alias="target", value=target.target), ], ) @@ -160,7 +160,7 @@ def tosa_formatters(data: Any) -> Callable[[Any], Report]: return report_advice if isinstance(data, TOSAConfiguration): - return report_device + return report_target if isinstance(data, MetadataDisplay): return report_metadata |