From 1ee2eb57342b6c9364803f8b557cb56e8ef5cefe Mon Sep 17 00:00:00 2001 From: ontocord Date: Tue, 1 Mar 2022 13:12:33 -0500 Subject: [PATCH] Update text_augment.py --- text_augment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/text_augment.py b/text_augment.py index 3217993..9d4717b 100644 --- a/text_augment.py +++ b/text_augment.py @@ -1675,7 +1675,7 @@ def process_ner_chunks_with_trans(self, target_offset_key = f'{target_lang}_offset' target_src_sim_key = f'{src_lang}_2_{target_lang}_sim' - public_figure_kenlm_data_list = public_figure_kenlm_cutoff_map.get(target_lang, {'cutoff': 500, 'pattern': "{} (born"}) + public_figure_kenlm_data_list = public_figure_kenlm_cutoff_map.get(target_lang, [{'cutoff': 500, 'pattern': "{} (born"}]) docs = self.collapse_ner(docs, ner_key = f'{src_lang}_signal_ner', collapse_ner_key = f'{src_lang}_ner', text_key = f'{src_lang}_text', stopwords=stopwords1) # do operations in the target_lang space