diff options
-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) |