diff --git a/model_compression_toolkit/core/runner.py b/model_compression_toolkit/core/runner.py index ac5db54bd..69b658a12 100644 --- a/model_compression_toolkit/core/runner.py +++ b/model_compression_toolkit/core/runner.py @@ -90,7 +90,7 @@ def core_runner(in_model: Any, # Checking whether to run mixed precision quantization if target_resource_utilization is not None and target_resource_utilization.is_any_restricted(): - if core_config.mixed_precision_config is None: + if core_config.mixed_precision_config is None: # pragma: no cover Logger.critical("Provided an initialized target_resource_utilization, that means that mixed precision quantization is " "enabled, but the provided MixedPrecisionQuantizationConfig is None.") if target_resource_utilization.activation_restricted() or target_resource_utilization.total_mem_restricted(): diff --git a/tests/keras_tests/feature_networks_tests/feature_networks/manual_bit_selection.py b/tests/keras_tests/feature_networks_tests/feature_networks/manual_bit_selection.py index 1ea3b589d..43f030d3d 100644 --- a/tests/keras_tests/feature_networks_tests/feature_networks/manual_bit_selection.py +++ b/tests/keras_tests/feature_networks_tests/feature_networks/manual_bit_selection.py @@ -208,3 +208,7 @@ def get_tpc(self): def get_resource_utilization(self): return mct.core.ResourceUtilization(activation_memory=6000) + + def compare(self, quantized_model, float_model, input_x=None, quantization_info=None): + self.unit_test.assertTrue(len(quantization_info.mixed_precision_cfg) > 0, "Expected mixed-precision in test.") + super().compare(quantized_model, float_model, input_x=input_x, quantization_info=quantization_info)