diff options
| author | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2023-08-24 16:38:47 +0100 |
|---|---|---|
| committer | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2023-09-05 14:20:08 +0100 |
| commit | e5a0bc3ecd4d9c46ead3b8217584eaa916a3afa4 (patch) | |
| tree | 94c348fcef50326a755a049a2a4027f588211f8b /tests/test_nn_tensorflow_config.py | |
| parent | 900c3c52b681e0b8a4454e2e2cf29265d53a2c98 (diff) | |
| download | mlia-e5a0bc3ecd4d9c46ead3b8217584eaa916a3afa4.tar.gz | |
MLIA-961 Update tox dependencies
- Update version dependencies in the tox.ini
- Fix linter issues
Change-Id: I04c3a841ee2646a865dab037701d66c28792f2a4
Signed-off-by: Benjamin Klimczak <benjamin.klimczak@arm.com>
Diffstat (limited to 'tests/test_nn_tensorflow_config.py')
| -rw-r--r-- | tests/test_nn_tensorflow_config.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/tests/test_nn_tensorflow_config.py b/tests/test_nn_tensorflow_config.py index 1a6fbe3..656619d 100644 --- a/tests/test_nn_tensorflow_config.py +++ b/tests/test_nn_tensorflow_config.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 config module.""" from contextlib import ExitStack as does_not_raise @@ -45,10 +45,12 @@ def test_convert_tf_to_tflite(tmp_path: Path, test_tf_model: Path) -> None: "test.model", None, pytest.raises( - Exception, - match="The input model format is not supported" - r"\(supported formats: TensorFlow Lite, Keras, " - r"TensorFlow saved model\)!", + ValueError, + match=( + "The input model format is not supported " + r"\(supported formats: TensorFlow Lite, Keras, " + r"TensorFlow saved model\)!" + ), ), ), ], |