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_optimizations_select.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_optimizations_select.py')
-rw-r--r-- | tests/test_nn_tensorflow_optimizations_select.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/tests/test_nn_tensorflow_optimizations_select.py b/tests/test_nn_tensorflow_optimizations_select.py index e22a9d8..f5ba6f0 100644 --- a/tests/test_nn_tensorflow_optimizations_select.py +++ b/tests/test_nn_tensorflow_optimizations_select.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 module select.""" from __future__ import annotations @@ -10,6 +10,7 @@ from typing import Any import pytest import tensorflow as tf +from mlia.core.errors import ConfigurationError from mlia.nn.tensorflow.optimizations.clustering import Clusterer from mlia.nn.tensorflow.optimizations.clustering import ClusteringConfiguration from mlia.nn.tensorflow.optimizations.pruning import Pruner @@ -55,7 +56,7 @@ from mlia.nn.tensorflow.optimizations.select import OptimizationSettings layers_to_optimize=None, ), pytest.raises( - Exception, + ConfigurationError, match="Optimization target should be a " "positive integer. " "Optimization target provided: 0.5", @@ -76,7 +77,7 @@ from mlia.nn.tensorflow.optimizations.select import OptimizationSettings layers_to_optimize="all", # type: ignore ), pytest.raises( - Exception, + ConfigurationError, match="Unsupported optimization type: superoptimization", ), None, @@ -89,7 +90,7 @@ from mlia.nn.tensorflow.optimizations.select import OptimizationSettings layers_to_optimize=None, ), pytest.raises( - Exception, + ConfigurationError, match="Optimization type is not provided", ), None, @@ -225,7 +226,7 @@ def test_optimization_settings_create_from( OptimizationSettings("super_optimization", 42, None), None, pytest.raises( - Exception, match="Unknown optimization type super_optimization" + Exception, match="Optimization type super_optimization is unknown." ), ], ], |