From d1a0987a80add8a08c3f8410a12c6857de00eab6 Mon Sep 17 00:00:00 2001 From: mart-r Date: Fri, 28 Aug 2026 13:29:43 +0100 Subject: [PATCH 1/3] Update wrapper API in line with 2.9.0 core release --- medcat-den/src/medcat_den/wrappers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/medcat-den/src/medcat_den/wrappers.py b/medcat-den/src/medcat_den/wrappers.py index c3c76981c..972ffa844 100644 --- a/medcat-den/src/medcat_den/wrappers.py +++ b/medcat-den/src/medcat_den/wrappers.py @@ -1,10 +1,11 @@ -from typing import Union, Optional +from typing import Union, Optional, Type from medcat.cat import CAT from medcat.utils.defaults import DEFAULT_PACK_NAME from medcat.storage.serialisers import AvailableSerialisers from medcat.trainer import Trainer from medcat.data.mctexport import MedCATTrainerExport +from medcat.components.addons.addons import AddonComponent from medcat_den.base import ModelInfo from medcat_den.config import DenConfig, RemoteDenConfig @@ -92,6 +93,7 @@ def load_model_pack(cls, model_pack_path: str, addon_config_dict: Optional[dict[str, dict]] = None, model_info: Optional[ModelInfo] = None, den_cnf: Optional[DenConfig] = None, + keep_addons_of_types: Optional[list[Type[AddonComponent]]] = None, ) -> 'CAT': """Load the model pack from file. From 08f10e01886b0f490f91407fbcc31209fc6fd602 Mon Sep 17 00:00:00 2001 From: mart-r Date: Fri, 28 Aug 2026 13:30:53 +0100 Subject: [PATCH 2/3] Propagate addon loads to core lib --- medcat-den/src/medcat_den/wrappers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/medcat-den/src/medcat_den/wrappers.py b/medcat-den/src/medcat_den/wrappers.py index 972ffa844..090871335 100644 --- a/medcat-den/src/medcat_den/wrappers.py +++ b/medcat-den/src/medcat_den/wrappers.py @@ -123,7 +123,9 @@ def load_model_pack(cls, model_pack_path: str, CAT: The loaded model pack. """ _cat = super().load_model_pack( - model_pack_path, config_dict, addon_config_dict) + model_pack_path, config_dict, addon_config_dict, + keep_addons_of_types=keep_addons_of_types, + ) cat = cls(_cat) if model_info is None: raise CannotWrapModel("Model info must be provided") From 310ad3debe661e786b643bdb1912b11a4f76bcc7 Mon Sep 17 00:00:00 2001 From: mart-r Date: Fri, 28 Aug 2026 13:53:09 +0100 Subject: [PATCH 3/3] Move required argument up in the wrapper --- medcat-den/src/medcat_den/wrappers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/medcat-den/src/medcat_den/wrappers.py b/medcat-den/src/medcat_den/wrappers.py index 090871335..6bd09fcda 100644 --- a/medcat-den/src/medcat_den/wrappers.py +++ b/medcat-den/src/medcat_den/wrappers.py @@ -91,9 +91,9 @@ def trainer(self) -> Trainer: def load_model_pack(cls, model_pack_path: str, config_dict: Optional[dict] = None, addon_config_dict: Optional[dict[str, dict]] = None, + keep_addons_of_types: Optional[list[Type[AddonComponent]]] = None, model_info: Optional[ModelInfo] = None, den_cnf: Optional[DenConfig] = None, - keep_addons_of_types: Optional[list[Type[AddonComponent]]] = None, ) -> 'CAT': """Load the model pack from file.