diff options
author | Ruomei Yan <ruomei.yan@arm.com> | 2023-01-19 14:52:36 +0000 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2023-02-13 14:24:02 +0000 |
commit | e3bef50932abdc2e32fa5636fb7a496b149e72d9 (patch) | |
tree | d32e184038ef0cd9e206810abe6dd44d84067163 /src/mlia/target/tosa/advisor.py | |
parent | cceb6fbd548d6414928f779e5572d325beb3c604 (diff) | |
download | mlia-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.py | 9 |
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), ) ] |