Update text_augment.py

huu4ontocord · Mar 1, 2022 · 1ee2eb5 · 1ee2eb5
1 parent 6248c96
commit 1ee2eb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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