aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/target/tosa/metadata.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/metadata.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/metadata.py')
-rw-r--r--src/mlia/target/tosa/metadata.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/src/mlia/target/tosa/metadata.py b/src/mlia/target/tosa/metadata.py
index 5575207..8e1f5ca 100644
--- a/src/mlia/target/tosa/metadata.py
+++ b/src/mlia/target/tosa/metadata.py
@@ -2,7 +2,16 @@
# SPDX-License-Identifier: Apache-2.0
"""TOSA package metadata."""
from mlia.core.metadata import Metadata
+from mlia.utils.misc import get_pkg_version
class TOSAMetadata(Metadata): # pylint: disable=too-few-public-methods
"""TOSA metadata."""
+
+ def __init__(self) -> None:
+ """Init TOSAMetadata."""
+ super().__init__("tosa-checker")
+
+ def get_metadata(self) -> dict:
+ """Returen tosa version."""
+ return {"tosa_version": get_pkg_version(self.name)}