Temporary fix to #43

kevinlu1248 · Jun 28, 2021 · 2ec1fc0 · 2ec1fc0
1 parent e00d632
commit 2ec1fc0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/pyate/term_extraction.py b/src/pyate/term_extraction.py
@@ -247,7 +247,7 @@ def add_to_counter(matcher, doc, i, matches):
             for end_index, (insert_order, original_value) in self.trie.iter(
                     document.lower()):
                 term_counter[original_value] += 1
-        return term_counter
+        return dict(term_counter)
 
     def count_terms_from_documents(self,
                                    seperate: bool = False,
@@ -260,6 +260,7 @@ def count_terms_from_documents(self,
         """
         # TODO: further optimize
         # TODO: add type annotations
+
         if hasattr(self, "_TermExtraction__term_counts"):
             return self.__term_counts
 
@@ -289,7 +290,7 @@ def callback(counter_dict):
                         term_counter[term] += frequency
 
             def error_callback(e):
-                print("Error: " + e)
+                print("Error: " + str(e))
 
             if doAsync:
                 P = Pool()