diff options
| author | Dhruv Chauhan <dhruv.chauhan@arm.com> | 2023-04-03 12:23:57 +0100 |
|---|---|---|
| committer | Dhruv Chauhan <dhruv.chauhan@arm.com> | 2023-04-04 14:39:13 +0100 |
| commit | c7c894ee0721b1076fcea25506d7b84772cc5704 (patch) | |
| tree | df42702d6c486e2193cafa00cec544c3182ecc7b | |
| parent | edf436c48029aa4e2b4ca5d17eee5a8f07ecbd6f (diff) | |
| download | mlia-c7c894ee0721b1076fcea25506d7b84772cc5704.tar.gz | |
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
| -rw-r--r-- | src/mlia/target/tosa/metadata.py | 4 | ||||
| -rw-r--r-- | 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) |