changed copy_from_cache

supervisely-ecosystem · Feb 28, 2024 · 0b99ef9 · 0b99ef9
1 parent 173006a
commit 0b99ef9
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/train/src/dataset_cache.py b/train/src/dataset_cache.py
@@ -64,10 +64,10 @@ def download_project(
     # copy datasets from cache
     total = sum([sly.get_cache_size(project_info.id, ds.name) for ds in dataset_infos])
     with progress(message="Retreiving data from cache...", total=total) as pbar:
-        for ds_info in dataset_infos:
-            sly.copy_from_cache(
-                project_id=project_info.id,
-                dest_dir=g.project_dir,
-                dataset_name=ds_info.name,
-                progress_cb=pbar.update
-            )
+        dataset_names = [ds_info.name for ds_info in dataset_infos]
+        sly.copy_from_cache(
+            project_id=project_info.id,
+            dest_dir=g.project_dir,
+            dataset_names=dataset_names,
+            progress_cb=pbar.update
+        )