Bugfix in NerModelConfiguration (#1230)

microsoft · Dec 21, 2023 · a6e253d · a6e253d
1 parent e7197da
commit a6e253d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/presidio-analyzer/presidio_analyzer/nlp_engine/ner_model_configuration.py b/presidio-analyzer/presidio_analyzer/nlp_engine/ner_model_configuration.py
@@ -26,7 +26,7 @@
     HOSPITAL="ORGANIZATION",
 )
 
-LOW_SCORE_ENTITY_NAMES = {}
+LOW_SCORE_ENTITY_NAMES = set()
 LABELS_TO_IGNORE = {
     "O",
     "ORG",
@@ -94,12 +94,12 @@ def __post_init__(self):
     @classmethod
     def _validate_input(cls, ner_model_configuration_dict: Dict) -> None:
         key_to_type = {
-            "labels_to_ignore": list,
+            "labels_to_ignore": Collection,
             "aggregation_strategy": str,
             "alignment_mode": str,
             "model_to_presidio_entity_mapping": dict,
             "low_confidence_score_multiplier": float,
-            "low_score_entity_names": list,
+            "low_score_entity_names": Collection,
             "stride": int,
         }