From d3f37f798e0ecd03a9c360d87c744a8481ad9612 Mon Sep 17 00:00:00 2001 From: Shahzaib Ahmad Date: Sat, 22 Aug 2026 20:56:15 +0500 Subject: [PATCH] Fix HuggingFace NER kwargs handling (#1188) Co-authored-by: Shahzaib Ahmad Co-authored-by: Zohaib Hassnain <109234410+ZohaibHassan16@users.noreply.github.com> --- semantica/semantic_extract/methods.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/semantica/semantic_extract/methods.py b/semantica/semantic_extract/methods.py index f0559dec..e7ac62b2 100644 --- a/semantica/semantic_extract/methods.py +++ b/semantica/semantic_extract/methods.py @@ -779,9 +779,11 @@ def extract_entities_huggingface( """ loader = HuggingFaceModelLoader(device=device) # Pass kwargs (like aggregation_strategy) to load_ner_model - model_obj = loader.load_ner_model(model, **kwargs) + loader_kwargs = { + key: value for key, value in kwargs.items() if key != "huggingface_model" + } + model_obj = loader.load_ner_model(model, **loader_kwargs) results = loader.extract_entities(model_obj, text) - entities = [] # Check if manual aggregation is needed (raw IOB tags detected)