ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 17:17:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix Logger.report_text() on Task created using Task.created() was not supported

Browse Source
This commit is contained in:
allegroai 2020-11-25 11:02:26 +02:00
parent 32c134eaa7
commit 313e99f156
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
trains/backend_interface/task/log.py
View File

@ -29,7 +29,7 @@ class TaskHandler(BufferingHandler):
def task_id(self, value): def task_id(self, value):
self._task_id = value self._task_id = value
def __init__(self, task, capacity=buffer_capacity): def __init__(self, task, capacity=buffer_capacity, connect_logger=True):
super(TaskHandler, self).__init__(capacity) super(TaskHandler, self).__init__(capacity)
self.task_id = task.id self.task_id = task.id
self.session = task.session self.session = task.session
@ -41,6 +41,7 @@ class TaskHandler(BufferingHandler):
self._thread = None self._thread = None
self._pending = 0 self._pending = 0
self._offline_log_filename = None self._offline_log_filename = None
self._connect_logger = connect_logger
if task.is_offline(): if task.is_offline():
offline_folder = Path(task.get_offline_mode_folder()) offline_folder = Path(task.get_offline_mode_folder())
offline_folder.mkdir(parents=True, exist_ok=True) offline_folder.mkdir(parents=True, exist_ok=True)
@ -58,7 +59,7 @@ class TaskHandler(BufferingHandler):
# if we need to add handlers to the base_logger, # if we need to add handlers to the base_logger,
# it will not automatically create stream one when first used, so we must manually configure it. # it will not automatically create stream one when first used, so we must manually configure it.
if not TaskHandler.__once: if self._connect_logger and not TaskHandler.__once:
base_logger = getLogger() base_logger = getLogger()
if len(base_logger.handlers) == 1 and isinstance(base_logger.handlers[0], TaskHandler): if len(base_logger.handlers) == 1 and isinstance(base_logger.handlers[0], TaskHandler):
if record.name != 'console' and not record.name.startswith('trains.'): if record.name != 'console' and not record.name.startswith('trains.'):