diff options
Diffstat (limited to 'tests/test_nn_select.py')
-rw-r--r-- | tests/test_nn_select.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/tests/test_nn_select.py b/tests/test_nn_select.py index 92b7a3d..15abf2d 100644 --- a/tests/test_nn_select.py +++ b/tests/test_nn_select.py @@ -143,19 +143,19 @@ from mlia.nn.tensorflow.optimizations.pruning import PruningConfiguration ( OptimizationSettings( optimization_type="rewrite", - optimization_target="fully_connected", # type: ignore + optimization_target="fully-connected", # type: ignore layers_to_optimize=None, dataset=None, ), does_not_raise(), RewritingOptimizer, - "rewrite: fully_connected", + "rewrite: fully-connected", ), ( - RewriteConfiguration("fully_connected"), + RewriteConfiguration("fully-connected"), does_not_raise(), RewritingOptimizer, - "rewrite: fully_connected", + "rewrite: fully-connected", ), ], ) @@ -192,7 +192,7 @@ def test_get_optimizer_training_parameters( """Test function get_optimzer with various combinations of parameters.""" config = OptimizationSettings( optimization_type="rewrite", - optimization_target="fully_connected", # type: ignore + optimization_target="fully-connected", # type: ignore layers_to_optimize=None, dataset=None, ) |