add log level as an argument to create logger

surfnerd · Mar 4, 2020 · e267dbb · e267dbb
1 parent 6fa4190
commit e267dbb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/ml-agents/mlagents/logging_util.py b/ml-agents/mlagents/logging_util.py
@@ -1,8 +1,7 @@
 import logging
 
 
-def create_logger(name):
-    log_level = logging.INFO
+def create_logger(name, log_level=logging.INFO):
     date_format = "%Y-%m-%d %H:%M:%S"
     log_format = "%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] %(message)s"
 

diff --git a/ml-agents/mlagents/trainers/learn.py b/ml-agents/mlagents/trainers/learn.py
@@ -451,16 +451,14 @@ def run_cli(options: RunOptions) -> None:
     except Exception:
         print("\n\n\tUnity Technologies\n")
     print(get_version_string())
-    trainer_logger = create_logger("mlagents.trainers")
-    env_logger = create_logger("mlagents_envs")
+    trainer_logger = create_logger("mlagents.trainers", log_level=logging.INFO)
+    env_logger = create_logger("mlagents_envs", log_level=logging.INFO)
 
     if options.debug:
         trainer_logger.setLevel("DEBUG")
         env_logger.setLevel("DEBUG")
     else:
-        trainer_logger.setLevel("INFO")
-        env_logger.setLevel("INFO")
-        # disable noisy warnings from tensorflow.
+        # disable noisy warnings from tensorflow
         tf_utils.set_warnings_enabled(False)
 
     trainer_logger.debug("Configuration for this run:")