Fix getting datasets fails

This commit is contained in:
allegroai 2022-07-21 17:24:54 +03:00
parent 51f512889f
commit 0397f2b41e

View File

@ -165,16 +165,17 @@ class Dataset(object):
for key in {'files added', 'files removed', 'files modified'}} for key in {'files added', 'files removed', 'files modified'}}
else: else:
self.changed_files = {'files added': 0, 'files removed': 0, 'files modified': 0} self.changed_files = {'files added': 0, 'files removed': 0, 'files modified': 0}
dataset_project, parent_project = self._build_hidden_project_name(task.get_project_name(), task.name) if "/.datasets/" not in task.get_project_name() or "":
task.move_to_project(new_project_name=dataset_project) dataset_project, parent_project = self._build_hidden_project_name(task.get_project_name(), task.name)
if bool(Session.check_min_api_server_version(Dataset.__min_api_version)): task.move_to_project(new_project_name=dataset_project)
get_or_create_project(task.session, project_name=parent_project, system_tags=[self.__hidden_tag]) if bool(Session.check_min_api_server_version(Dataset.__min_api_version)):
get_or_create_project( get_or_create_project(task.session, project_name=parent_project, system_tags=[self.__hidden_tag])
task.session, get_or_create_project(
project_name=dataset_project, task.session,
project_id=task.project, project_name=dataset_project,
system_tags=[self.__hidden_tag, self.__tag], project_id=task.project,
) system_tags=[self.__hidden_tag, self.__tag],
)
else: else:
self._created_task = True self._created_task = True
dataset_project, parent_project = self._build_hidden_project_name(dataset_project, dataset_name) dataset_project, parent_project = self._build_hidden_project_name(dataset_project, dataset_name)
@ -1251,7 +1252,7 @@ class Dataset(object):
dataset_version=None, # Optional[str] dataset_version=None, # Optional[str]
entire_dataset=False, # bool entire_dataset=False, # bool
action=None, # Optional[str] action=None, # Optional[str]
shallow_search=True, # bool shallow_search=False, # bool
): ):
# type: (...) -> List[str] # type: (...) -> List[str]
""" """
@ -1316,7 +1317,7 @@ class Dataset(object):
force=False, # bool force=False, # bool
dataset_version=None, # Optional[str] dataset_version=None, # Optional[str]
entire_dataset=False, # bool entire_dataset=False, # bool
shallow_search=True # bool shallow_search=False # bool
): ):
# type: (...) -> () # type: (...) -> ()
""" """
@ -1478,7 +1479,7 @@ class Dataset(object):
dataset_version=None, # type: Optional[str] dataset_version=None, # type: Optional[str]
alias=None, # type: Optional[str] alias=None, # type: Optional[str]
overridable=False, # type: bool overridable=False, # type: bool
shallow_search=True, # type: bool shallow_search=False, # type: bool
**kwargs **kwargs
): ):
# type: (...) -> "Dataset" # type: (...) -> "Dataset"