aboutsummaryrefslogtreecommitdiff
path: root/tests/test_target_tosa_advisor.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_target_tosa_advisor.py')
-rw-r--r--tests/test_target_tosa_advisor.py21
1 files changed, 6 insertions, 15 deletions
diff --git a/tests/test_target_tosa_advisor.py b/tests/test_target_tosa_advisor.py
index 020acc5..d0b42b9 100644
--- a/tests/test_target_tosa_advisor.py
+++ b/tests/test_target_tosa_advisor.py
@@ -9,6 +9,7 @@ import pytest
from mlia.core.common import AdviceCategory
from mlia.core.context import ExecutionContext
from mlia.core.workflow import DefaultWorkflowExecutor
+from mlia.target.common.optimization import _DEFAULT_OPTIMIZATION_TARGETS
from mlia.target.tosa.advisor import configure_and_get_tosa_advisor
from mlia.target.tosa.advisor import TOSAInferenceAdvisor
@@ -33,21 +34,11 @@ def test_configure_and_get_tosa_advisor(
assert ctx.event_handlers is not None
assert ctx.config_parameters == {
"common_optimizations": {
- "optimizations": [
- [
- {
- "layers_to_optimize": None,
- "optimization_target": 0.5,
- "optimization_type": "pruning",
- },
- {
- "layers_to_optimize": None,
- "optimization_target": 32,
- "optimization_type": "clustering",
- },
- ]
- ],
- "training_parameters": None,
+ "optimizations": [_DEFAULT_OPTIMIZATION_TARGETS],
+ "rewrite_parameters": {
+ "train_params": None,
+ "rewrite_specific_params": None,
+ },
},
"tosa_inference_advisor": {
"model": str(test_tflite_model),