Download all nested datasets when running from specific dataset (#22)

supervisely-ecosystem · Dec 2, 2024 · 5a6df61 · 5a6df61
1 parent 14f1d8a
commit 5a6df61
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/main.py b/src/main.py
@@ -107,6 +107,9 @@ def download(project: sly.Project) -> str:
 
     if dataset_id is not None:
         dataset_ids = [dataset_id]
+        nested_datasets = api.dataset.get_nested(project.id, dataset_id)
+        nested_dataset_ids = [dataset.id for dataset in nested_datasets]
+        dataset_ids.extend(nested_dataset_ids)
     else:
         datasets = api.dataset.get_list(project.id, recursive=True)
         dataset_ids = [dataset.id for dataset in datasets]