diff --git a/mlair/run_modules/run_environment.py b/mlair/run_modules/run_environment.py index 5414b21cb0cb26674c699a02c22400959e11f1aa..df34345b4fb67e764f6e4d8d6570a5fafb762304 100644 --- a/mlair/run_modules/run_environment.py +++ b/mlair/run_modules/run_environment.py @@ -92,12 +92,12 @@ class RunEnvironment(object): logger = None tracker_list = [] - def __init__(self, name=None): + def __init__(self, name=None, log_level_stream=None): """Start time tracking automatically and logs as info.""" if RunEnvironment.data_store is None: RunEnvironment.data_store = DataStoreObject() if RunEnvironment.logger is None: - RunEnvironment.logger = Logger() + RunEnvironment.logger = Logger(level_stream=log_level_stream) self._name = name if name is not None else self.__class__.__name__ self.time = TimeTracking(name=name) logging.info(f"{self._name} started")