Minor PR fixes

sony · Mar 11, 2024 · 83a0081 · 83a0081
1 parent 4643017
commit 83a0081
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/model_compression_toolkit/core/common/mixed_precision/search_methods/linear_programming.py b/model_compression_toolkit/core/common/mixed_precision/search_methods/linear_programming.py
@@ -233,7 +233,7 @@ def _add_set_of_kpi_constraints(search_manager: MixedPrecisionSearchManager,
 
 def _build_layer_to_metrics_mapping(search_manager: MixedPrecisionSearchManager,
                                     target_kpi: KPI,
-                                    eps: float = 1e-8) -> Dict[int, Dict[int, float]]:
+                                    eps: float = EPS) -> Dict[int, Dict[int, float]]:
     """
     This function measures the sensitivity of a change in a bitwidth of a layer on the entire model.
     It builds a mapping from a node's index, to its bitwidht's effect on the model sensitivity.

diff --git a/tests/keras_tests/non_parallel_tests/test_lp_search_bitwidth.py b/tests/keras_tests/non_parallel_tests/test_lp_search_bitwidth.py
@@ -38,7 +38,7 @@
 from model_compression_toolkit.target_platform_capabilities.tpc_models.imx500_tpc.latest import \
     get_op_quantization_configs
 from tests.keras_tests.tpc_keras import get_weights_only_mp_tpc_keras
-from pulp import *
+from pulp import lpSum
 
 class MockReconstructionHelper:
     def __init__(self):