ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-12 07:35:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix TaskScheduler does not save local state before running remotely

Browse Source
This commit is contained in:
allegroai 2024-01-18 17:09:42 +02:00
parent a2d1ff18d0
commit 1b80a77c92
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/automation/scheduler.py
View File

@ -385,7 +385,13 @@ class BaseScheduler(object):
:param queue: Remote queue to run the scheduler on, default 'services' queue. :param queue: Remote queue to run the scheduler on, default 'services' queue.
""" """
# make sure we serialize the current state if we are running locally
if self._task.running_locally():
self._serialize_state()
self._serialize()
# launch on the remote agent
self._task.execute_remotely(queue_name=queue, exit_process=True) self._task.execute_remotely(queue_name=queue, exit_process=True)
# we will be deserializing the state inside `start`
self.start() self.start()
def _update_execution_plots(self): def _update_execution_plots(self):