ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix duplicate task dataset tags

Browse Source
This commit is contained in:
allegroai 2021-10-16 23:04:11 +03:00
parent f986c72113
commit 9085847c6d
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml/datasets/dataset.py
View File

@ -84,7 +84,9 @@ class Dataset(object):
if str(task_status) in ('created', 'in_progress'): if str(task_status) in ('created', 'in_progress'):
if str(task.task_type) != str(Task.TaskTypes.data_processing): if str(task.task_type) != str(Task.TaskTypes.data_processing):
task.set_task_type(task_type=Task.TaskTypes.data_processing) task.set_task_type(task_type=Task.TaskTypes.data_processing)
task.set_system_tags((task.get_system_tags() or []) + [self.__tag]) task_system_tags = task.get_system_tags() or []
if self.__tag not in task_system_tags:
task.set_system_tags(task_system_tags + [self.__tag])
if dataset_tags: if dataset_tags:
task.set_tags((task.get_tags() or []) + list(dataset_tags)) task.set_tags((task.get_tags() or []) + list(dataset_tags))
else: else: