From e3bef50932abdc2e32fa5636fb7a496b149e72d9 Mon Sep 17 00:00:00 2001 From: Ruomei Yan Date: Thu, 19 Jan 2023 14:52:36 +0000 Subject: 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 --- src/mlia/target/tosa/metadata.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'src/mlia/target/tosa/metadata.py') 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)} -- cgit v1.2.1