add comp back

generic_ner.py

@@ -736,7 +736,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)
@@ -752,13 +752,17 @@ 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)
+        if DEBUG:
+            print("After remove_included_entities:", all_entities)
         all_entities = remove_trailing_stopwords(all_entities)
-        print("After remove_trailing_stopwords:", all_entities)
+        if DEBUG:
+            print("After remove_trailing_stopwords:", all_entities)
         all_entities = postprocess_entities(all_entities)
-        print("After postprocess_entities:", all_entities)
+        if DEBUG:
+            print("After postprocess_entities:", all_entities)
         all_entities = refine_entities_with_coarse(all_entities, coarse_entities)
-        print("After refine_entities_with_coarse:", all_entities)
+        if DEBUG:
+            print("After refine_entities_with_coarse:", all_entities)
         # print("After attach_comp_to_closest:")
         # pprint(all_entities)
         # print("\n")