From f7f53676f412b46b934d79bb7be0243ab2a54e58 Mon Sep 17 00:00:00 2001 From: eladc4 Date: Sun, 29 Dec 2024 09:29:46 +0200 Subject: [PATCH] Fix bug --- .../common/mixed_precision/mixed_precision_search_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_compression_toolkit/core/common/mixed_precision/mixed_precision_search_manager.py b/model_compression_toolkit/core/common/mixed_precision/mixed_precision_search_manager.py index a6d908d8e..7fbb0807b 100644 --- a/model_compression_toolkit/core/common/mixed_precision/mixed_precision_search_manager.py +++ b/model_compression_toolkit/core/common/mixed_precision/mixed_precision_search_manager.py @@ -42,7 +42,7 @@ def __init__(self, fw_info: FrameworkInfo, fw_impl: FrameworkImplementation, sensitivity_evaluator: SensitivityEvaluation, - ru_functions: Dict[RUTarget, RuFunctions[MpRuMetric, MpRuAggregation]], + ru_functions: Dict[RUTarget, RuFunctions], target_resource_utilization: ResourceUtilization, original_graph: Graph = None): """