Skip to content

Commit 4e1bedb

Browse files
authored
Update anonymization.py
1 parent 2905aff commit 4e1bedb

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

ac_dc/anonymization.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ def apply_regex_anonymization(
88
lang_id: str,
99
context_window: int = 20,
1010
anonymize_condition=None,
11-
tag_type=None,
11+
tag_type= {'IP_ADDRESS', 'KEY', 'ID', 'PHONE', 'USER', 'EMAIL', 'LICENSE_PLATE'},
1212
) -> str:
1313
"""
1414
Params:
@@ -17,7 +17,7 @@ def apply_regex_anonymization(
1717
lang_id: str, the language id of the sentence
1818
context_window: int, the context window size
1919
anonymize_condition: function, the anonymization condition
20-
tag_type: iterable, the tag types of the anonymization. All keys in regex_rulebase by default
20+
tag_type: iterable, the tag types of the anonymization. All keys in regex_rulebase is None
2121
"""
2222
if tag_type == None:
2323
tag_type = regex_rulebase.keys()

0 commit comments

Comments
 (0)