From 597a7ed05e2376ec48604465cf5ebd752cebae9c Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Sun, 5 Sep 2021 00:28:20 +0300 Subject: [PATCH] Add Dataset create set parent current Task --- clearml/datasets/dataset.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/clearml/datasets/dataset.py b/clearml/datasets/dataset.py index 0a67b806..b97f878b 100644 --- a/clearml/datasets/dataset.py +++ b/clearml/datasets/dataset.py @@ -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