aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/target/tosa/advisor.py
diff options
context:
space:
mode:
authorRuomei Yan <ruomei.yan@arm.com>2023-01-19 14:52:36 +0000
committerRuomei Yan <ruomei.yan@arm.com>2023-02-13 14:24:02 +0000
commite3bef50932abdc2e32fa5636fb7a496b149e72d9 (patch)
treed32e184038ef0cd9e206810abe6dd44d84067163 /src/mlia/target/tosa/advisor.py
parentcceb6fbd548d6414928f779e5572d325beb3c604 (diff)
downloadmlia-e3bef50932abdc2e32fa5636fb7a496b149e72d9.tar.gz
MLIA-775 Refactor metadata related classes
- define Metadata base class with dictionary data and abstract method - mlia, tosa, model and metadatadisplay classes are all inherited from base class - update unit tests - update function report_metadata into more generalized format Change-Id: Id49e15283eebdca705045eda81db637d82f85453
Diffstat (limited to 'src/mlia/target/tosa/advisor.py')
-rw-r--r--src/mlia/target/tosa/advisor.py9
1 files changed, 1 insertions, 8 deletions
diff --git a/src/mlia/target/tosa/advisor.py b/src/mlia/target/tosa/advisor.py
index 5fb18ed..7666df4 100644
--- a/src/mlia/target/tosa/advisor.py
+++ b/src/mlia/target/tosa/advisor.py
@@ -16,8 +16,6 @@ from mlia.core.context import ExecutionContext
from mlia.core.data_analysis import DataAnalyzer
from mlia.core.data_collection import DataCollector
from mlia.core.events import Event
-from mlia.core.metadata import MLIAMetadata
-from mlia.core.metadata import ModelMetadata
from mlia.target.registry import profile
from mlia.target.tosa.advice_generation import TOSAAdviceProducer
from mlia.target.tosa.config import TOSAConfiguration
@@ -25,7 +23,6 @@ from mlia.target.tosa.data_analysis import TOSADataAnalyzer
from mlia.target.tosa.data_collection import TOSAOperatorCompatibility
from mlia.target.tosa.events import TOSAAdvisorStartedEvent
from mlia.target.tosa.handlers import TOSAEventHandler
-from mlia.target.tosa.metadata import TOSAMetadata
from mlia.target.tosa.reporters import MetadataDisplay
@@ -69,11 +66,7 @@ class TOSAInferenceAdvisor(DefaultInferenceAdvisor):
TOSAAdvisorStartedEvent(
model,
cast(TOSAConfiguration, profile(target_profile)),
- MetadataDisplay(
- TOSAMetadata("tosa-checker"),
- MLIAMetadata("mlia"),
- ModelMetadata(model),
- ),
+ MetadataDisplay(model),
)
]