diff --git a/trains/binding/frameworks/__init__.py b/trains/binding/frameworks/__init__.py index f10634a9..711000b5 100644 --- 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 index c486f341..ca6de0ba 100644 --- 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