diff options
Diffstat (limited to 'tests/test_target_tosa_reporters.py')
-rw-r--r-- | tests/test_target_tosa_reporters.py | 4 |
1 files changed, 3 insertions, 1 deletions
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) |