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/metadata.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/metadata.py')
-rw-r--r-- | src/mlia/target/tosa/metadata.py | 9 |
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)} |