ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-04 02:57:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix Task.get_tasks() return type to Task

Browse Source
This commit is contained in:
allegroai 2020-05-08 22:07:43 +03:00
parent a5ff2ba9c8
commit f90f8f06e2
1 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
trains/task.py
View File

@ -2144,16 +2144,11 @@ class Task(_Task):
if task_ids: if task_ids:
if isinstance(task_ids, six.string_types): if isinstance(task_ids, six.string_types):
task_ids = [task_ids] task_ids = [task_ids]
return [ return [cls(private=cls.__create_protection, task_id=task_id, log_to_backend=False)
cls(private=cls.__create_protection, task_id=task, log_to_backend=False) for task_id in task_ids]
for task in task_ids
]
return cls._query_tasks( return [cls(private=cls.__create_protection, task_id=task.id, log_to_backend=False)
project_name=project_name, for task in cls._query_tasks(project_name=project_name, task_name=task_name, **kwargs)]
task_name=task_name,
**kwargs
)
@classmethod @classmethod
def _query_tasks(cls, task_ids=None, project_name=None, task_name=None, **kwargs): def _query_tasks(cls, task_ids=None, project_name=None, task_name=None, **kwargs):