diff options
| author | Gergely Nagy <gergely.nagy@arm.com> | 2023-11-21 12:29:38 +0000 |
|---|---|---|
| committer | Gergely Nagy <gergely.nagy@arm.com> | 2023-12-07 17:09:31 +0000 |
| commit | 54eec806272b7574a0757c77a913a369a9ecdc70 (patch) | |
| tree | 2e6484b857b2a68279a2707dbb21e5c26685f4e2 /tests/test_nn_tensorflow_tflite_compat.py | |
| parent | 7c50f1d6367186c03a282ac7ecb8fca0f905ba30 (diff) | |
| download | mlia-54eec806272b7574a0757c77a913a369a9ecdc70.tar.gz | |
MLIA-835 Invalid JSON output
TFLiteConverter was producing log messages in the output that was not
possible to capture and redirect to logging.
The solution/workaround is to run it as a subprocess.
This change required some refactoring around existing invocations of
the converter.
Change-Id: I394bd0d49d36e6686cfcb9d658e4aad05326cb87
Signed-off-by: Gergely Nagy <gergely.nagy@arm.com>
Diffstat (limited to 'tests/test_nn_tensorflow_tflite_compat.py')
| -rw-r--r-- | tests/test_nn_tensorflow_tflite_compat.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_nn_tensorflow_tflite_compat.py b/tests/test_nn_tensorflow_tflite_compat.py index f203125..4ca387c 100644 --- a/tests/test_nn_tensorflow_tflite_compat.py +++ b/tests/test_nn_tensorflow_tflite_compat.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Tests for tflite_compat module.""" from __future__ import annotations @@ -219,7 +219,7 @@ def test_tflite_compatibility( converter_mock.convert.side_effect = conversion_error monkeypatch.setattr( - "mlia.nn.tensorflow.tflite_compat.get_tflite_converter", + "mlia.nn.tensorflow.tflite_convert.get_tflite_converter", lambda *args, **kwargs: converter_mock, ) |