From e9b10e5f030b61c31863d7f510df316acba056e1 Mon Sep 17 00:00:00 2001 From: Pierre Chapuis Date: Mon, 27 Oct 2025 19:14:37 +0100 Subject: [PATCH] fix type hints in OptimizationConfig op_type_configs and op_name_configs are dicts (c.f. docstring above) --- coremltools/optimize/coreml/_config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/coremltools/optimize/coreml/_config.py b/coremltools/optimize/coreml/_config.py index e4958c439..87b60ddd9 100644 --- a/coremltools/optimize/coreml/_config.py +++ b/coremltools/optimize/coreml/_config.py @@ -837,8 +837,8 @@ class OptimizationConfig: * You can use ``coremltools.optimize.coreml.get_weights_metadata`` to get the name of the constants / op instances in the model. """ global_config: Optional[OpCompressorConfig] = field(default=None) - op_type_configs: Optional[OpCompressorConfig] = field(default=None) - op_name_configs: Optional[OpCompressorConfig] = field(default=None) + op_type_configs: Optional[Dict[str, OpCompressorConfig]] = field(default=None) + op_name_configs: Optional[Dict[str, OpCompressorConfig]] = field(default=None) # The following two private attributes is aim for backward compatibility for ct.compression_utils implementation # They need to be removed in the future once we deprecate ct.compression_utils