add comp back

generic_ner.py

@@ -743,7 +743,7 @@ class MultitaskTokenClassificationPipeline(Pipeline):
             entities[task] = get_entities(words_list, preds_list, confidence_list, text)
 
         # add titles to comp entities
-        # from pprint import pprint
+        from pprint import pprint
 
         # print("Before:")
         # pprint(entities)
@@ -759,13 +759,13 @@ class MultitaskTokenClassificationPipeline(Pipeline):
             print(all_entities)
         # print("After remove_included_entities:")
         all_entities = remove_included_entities(all_entities)
-        print("After remove_included_entities:", all_entities)
+        pprint("After remove_included_entities:", all_entities)
         all_entities = remove_trailing_stopwords(all_entities)
-        print("After remove_trailing_stopwords:", all_entities)
+        pprint("After remove_trailing_stopwords:", all_entities)
         all_entities = postprocess_entities(all_entities)
-        print("After postprocess_entities:", all_entities)
+        pprint("After postprocess_entities:", all_entities)
         all_entities = refine_entities_with_coarse(all_entities, coarse_entities)
-        print("After refine_entities_with_coarse:", all_entities)
+        pprint("After refine_entities_with_coarse:", all_entities)
         # print("After attach_comp_to_closest:")
         # pprint(all_entities)
         # print("\n")