From c7c894ee0721b1076fcea25506d7b84772cc5704 Mon Sep 17 00:00:00 2001 From: Dhruv Chauhan Date: Mon, 3 Apr 2023 12:23:57 +0100 Subject: MLIA-816 Correctly name TOSA Checker attribute - JSON output now reports tosa_checker as attribute name, which accurately represents its value as the version of TOSA Checker Change-Id: I8ac5032fea5dae65382db76091d295595a644816 --- src/mlia/target/tosa/metadata.py | 4 ++-- tests/test_target_tosa_reporters.py | 4 +++- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/mlia/target/tosa/metadata.py b/src/mlia/target/tosa/metadata.py index 8e1f5ca..603d6f7 100644 --- a/src/mlia/target/tosa/metadata.py +++ b/src/mlia/target/tosa/metadata.py @@ -13,5 +13,5 @@ class TOSAMetadata(Metadata): # pylint: disable=too-few-public-methods super().__init__("tosa-checker") def get_metadata(self) -> dict: - """Returen tosa version.""" - return {"tosa_version": get_pkg_version(self.name)} + """Return TOSA checker version.""" + return {"tosa_checker_version": get_pkg_version(self.name)} diff --git a/tests/test_target_tosa_reporters.py b/tests/test_target_tosa_reporters.py index 5f26c20..16d92ca 100644 --- a/tests/test_target_tosa_reporters.py +++ b/tests/test_target_tosa_reporters.py @@ -32,7 +32,9 @@ def test_tosa_formatters( display_data = MetadataDisplay(test_tflite_model) formatter = tosa_formatters(MetadataDisplay(test_tflite_model)) report = formatter(display_data) - assert display_data.data_dict["tosa-checker"]["tosa_version"] == mock_version + assert ( + display_data.data_dict["tosa-checker"]["tosa_checker_version"] == mock_version + ) assert isinstance(report, Report) -- cgit v1.2.1