diff options
Diffstat (limited to 'tests/test_cli_options.py')
| -rw-r--r-- | tests/test_cli_options.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tests/test_cli_options.py b/tests/test_cli_options.py index c02ef89..75ace0b 100644 --- a/tests/test_cli_options.py +++ b/tests/test_cli_options.py @@ -30,6 +30,7 @@ from mlia.core.typing import OutputFormat "optimization_type": "pruning", "optimization_target": 0.5, "layers_to_optimize": None, + "dataset": None, } ], ], @@ -44,6 +45,7 @@ from mlia.core.typing import OutputFormat "optimization_type": "pruning", "optimization_target": 0.5, "layers_to_optimize": None, + "dataset": None, } ], ], @@ -58,6 +60,7 @@ from mlia.core.typing import OutputFormat "optimization_type": "clustering", "optimization_target": 32, "layers_to_optimize": None, + "dataset": None, } ], ], @@ -72,11 +75,13 @@ from mlia.core.typing import OutputFormat "optimization_type": "pruning", "optimization_target": 0.5, "layers_to_optimize": None, + "dataset": None, }, { "optimization_type": "clustering", "optimization_target": 32, "layers_to_optimize": None, + "dataset": None, }, ], ], @@ -91,6 +96,7 @@ from mlia.core.typing import OutputFormat "optimization_type": "pruning", "optimization_target": 0.4, "layers_to_optimize": None, + "dataset": None, } ], ], @@ -117,6 +123,7 @@ from mlia.core.typing import OutputFormat "optimization_type": "clustering", "optimization_target": 32.2, "layers_to_optimize": None, + "dataset": None, } ], ], |