ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 18:52:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Dataset create set parent current Task

Browse Source
This commit is contained in:
allegroai 2021-09-05 00:28:20 +03:00
parent 6d09ff1518
commit 597a7ed05e
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
clearml/datasets/dataset.py
View File

@ -101,6 +101,9 @@ class Dataset(object):
# if the task is running make sure we ping to the server so it will not be aborted by a watchdog
self._task_pinger = DevWorker()
self._task_pinger.register(task, stop_signal_support=False)
# set the newly created Dataset parent ot the current Task, so we know who created it.
if Task.current_task() and Task.current_task().id != task.id:
task.set_parent(Task.current_task())
# store current dataset Task
self._task = task