diff --git a/presidio-analyzer/presidio_analyzer/nlp_engine/ner_model_configuration.py b/presidio-analyzer/presidio_analyzer/nlp_engine/ner_model_configuration.py index a88545090..8408f776a 100644 --- 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, }