From e5a0bc3ecd4d9c46ead3b8217584eaa916a3afa4 Mon Sep 17 00:00:00 2001 From: Benjamin Klimczak Date: Thu, 24 Aug 2023 16:38:47 +0100 Subject: MLIA-961 Update tox dependencies - Update version dependencies in the tox.ini - Fix linter issues Change-Id: I04c3a841ee2646a865dab037701d66c28792f2a4 Signed-off-by: Benjamin Klimczak --- tests/test_nn_tensorflow_optimizations_select.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'tests/test_nn_tensorflow_optimizations_select.py') 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." ), ], ], -- cgit v1.2.1