ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 18:52:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix Dataset.get dataset version without changeset

Browse Source
This commit is contained in:
allegroai 2021-01-11 18:29:34 +02:00
parent 582017f2d9
commit f1f595b886
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
clearml/datasets/dataset.py
View File

@ -985,6 +985,14 @@ class Dataset(object):
""" """
if not self._task: if not self._task:
self._task = Task.get_task(task_id=self._id) self._task = Task.get_task(task_id=self._id)
# check if we have a dataset with empty change set
if not self._task.artifacts.get(self.__data_entry_name):
cache = CacheManager.get_cache_manager(cache_context=self.__cache_context)
local_folder = Path(cache.get_cache_folder()) / self._get_cache_folder_name()
local_folder.mkdir(parents=True, exist_ok=True)
return local_folder.as_posix()
# download the dataset zip
local_zip = StorageManager.get_local_copy( local_zip = StorageManager.get_local_copy(
remote_url=self._task.artifacts[self.__data_entry_name].url, cache_context=self.__cache_context, remote_url=self._task.artifacts[self.__data_entry_name].url, cache_context=self.__cache_context,
extract_archive=False, name=self._id) extract_archive=False, name=self._id)