Fixed wrong code

PyThaiNLP · Apr 26, 2022 · 60fcd96 · 60fcd96
1 parent 6efee29
commit 60fcd96
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pythainlp/tag/lst20_ner_onnx.py b/pythainlp/tag/lst20_ner_onnx.py
@@ -23,10 +23,10 @@ def clean_output(self, list_text):
             elif i == '▁':
                 i = " "
             new_list.append((i, j))
-        return list_text
+        return new_list
 
     def _config(self, list_ner):
         _n = []
-        for i,j in list_ner:
-            _n.append((i,j.replace('E_', 'I_').replace('_', '-')))
+        for i, j in list_ner:
+            _n.append((i, j.replace('E_', 'I_').replace('_', '-')))
         return _n