Minor enhancements

EnzymeML · Mar 3, 2022 · cc2e8c1 · cc2e8c1
1 parent 677886d
commit cc2e8c1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pyenzyme/utils/log.py b/pyenzyme/utils/log.py
@@ -4,13 +4,13 @@
 from io import StringIO
 
 
-def setup_custom_logger(name, log_stream: StringIO):
+def setup_custom_logger(name, log_stream: StringIO, level = logging.DEBUG):
 
     formatter = logging.Formatter(
         fmt='%(levelname)s - %(message)s'
     )
 
-    handler = logging.StreamHandler(stream=sys.stdout)
+    handler = logging.StreamHandler(stream=sys.stderr)
     handler.setFormatter(formatter)
     handler.setLevel(logging.INFO)
 
@@ -23,10 +23,11 @@ def setup_custom_logger(name, log_stream: StringIO):
     string_handler.setLevel(logging.NOTSET)
 
     logger = logging.getLogger(name)
-    logger.setLevel(logging.DEBUG)
+    logger.setLevel(level)
     logger.addHandler(handler)
     logger.addHandler(string_handler)
 
+
     return logger