Allow Task.get_tasks() to filter by list of project ids

allegroai · Jun 21, 2020 · f3621fe · f3621fe
1 parent 2c5e2bb
commit f3621fe
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/trains/binding/frameworks/__init__.py b/trains/binding/frameworks/__init__.py
@@ -2,7 +2,6 @@
 import shutil
 import sys
 import threading
-import weakref
 from enum import Enum
 from random import randint
 from tempfile import mkstemp

diff --git a/trains/task.py b/trains/task.py
@@ -2330,7 +2330,7 @@ def _query_tasks(cls, task_ids=None, project_name=None, task_name=None, **kwargs
             cls._get_default_session(),
             tasks.GetAllRequest(
                 id=task_ids,
-                project=[project.id] if project else None,
+                project=[project.id] if project else kwargs.pop('project', None),
                 name=task_name if task_name else None,
                 only_fields=only_fields,
                 **kwargs