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

Allow setting multiprocessing start method

Browse Source
This commit is contained in:
Alex Burlacu 2023-07-31 21:34:13 +03:00
parent 80ef359f45
commit 92511efbde
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
clearml/backend_interface/task/repo/scriptinfo.py
View File

@ -273,8 +273,8 @@ class ScriptRequirements(object):
class _JupyterObserver(object): class _JupyterObserver(object):
_thread = None _thread = None
_exit_event = SafeEvent() _exit_event = None
_sync_event = SafeEvent() _sync_event = None
_sample_frequency = 30. _sample_frequency = 30.
_first_sample_frequency = 3. _first_sample_frequency = 3.
_jupyter_history_logger = None _jupyter_history_logger = None
@ -286,6 +286,10 @@ class _JupyterObserver(object):
@classmethod @classmethod
def observer(cls, jupyter_notebook_filename, notebook_name=None, log_history=False): def observer(cls, jupyter_notebook_filename, notebook_name=None, log_history=False):
if cls._exit_event is None:
cls._exit_event = SafeEvent()
if cls._sync_event is None:
cls._sync_event = SafeEvent()
if cls._thread is not None: if cls._thread is not None:
# order of signaling is important! # order of signaling is important!
cls._exit_event.set() cls._exit_event.set()
@ -304,6 +308,8 @@ class _JupyterObserver(object):
@classmethod @classmethod
def signal_sync(cls, *_, **__): def signal_sync(cls, *_, **__):
if cls._sync_event is None:
return
cls._sync_event.set() cls._sync_event.set()
@classmethod @classmethod