From 5f063ae1cfbfa2568d2858af0a0ccaf192bb1e8d Mon Sep 17 00:00:00 2001 From: Madeleine Dunn Date: Tue, 26 Mar 2024 13:20:42 +0000 Subject: fix: Update rewrite target name - Rename "fully_connected" to "fully-connected" - This will resolve issues with upstreaming rewrite library changes Signed-off-by: Madeleine Dunn Change-Id: I2f24ae4917a556fd0bd44f0db6ee4e0f7a68cd24 --- tests/test_nn_select.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'tests/test_nn_select.py') 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, ) -- cgit v1.2.1