Fixed: using logger instead of print

Author: nicolay-r

bulk_ner/annotate.py

@@ -65,9 +65,13 @@
 
     # Initialize NER model
     models_preset = {
-        "dynamic": lambda: dynamic_init(src_dir=CWD, class_filepath=ner_model_name, class_name=ner_model_params)(
+        "dynamic": lambda: dynamic_init(
+            src_dir=CWD, class_filepath=ner_model_name, class_name=ner_model_params,
+            handle_func=lambda class_path: logger.info((f"Dynamic loading for the file and class `{class_path}`"))
+        )(
             # The rest of parameters could be provided from cmd.
-            **model_args_dict)
+            **model_args_dict
+        )
     }
 
     annotator = NERAnnotator(ner_model=models_preset["dynamic"](),

bulk_ner/src/service_dynamic.py

@@ -2,13 +2,14 @@
 import sys
 
 
-def dynamic_init(src_dir, class_filepath, class_name=None):
+def dynamic_init(src_dir, class_filepath, class_name=None, handle_func=None):
     sys.path.append(src_dir)
     class_path_list = class_filepath.split('/')
     class_path_list[-1] = '.'.join(class_path_list[-1].split('.')[:-1])
     class_name = class_path_list[-1].title() if class_name is None else class_name
     class_path = ".".join(class_path_list + [class_name])
-    print(f"Dynamic loading for the file and class `{class_path}`")
+    if handle_func:
+        handle_func(class_path)
     cls = auto_import(class_path, is_class=False)
     return cls