wrong to_download datasets list fix

supervisely-ecosystem · Feb 28, 2024 · 510b3fe · 510b3fe
1 parent e32f5f4
commit 510b3fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train/src/dataset_cache.py b/train/src/dataset_cache.py
@@ -27,8 +27,8 @@ def download_project(
         return
 
     # get datasets to download and cached
-    to_download = [info for info in dataset_infos if sly.is_cached(project_info.id, info.name)]
-    cached = [info for info in dataset_infos if not sly.is_cached(project_info.id, info.name)]
+    to_download = [info for info in dataset_infos if not sly.is_cached(project_info.id, info.name)]
+    cached = [info for info in dataset_infos if sly.is_cached(project_info.id, info.name)]
     if len(cached) == 0:
         log_msg = "No cached datasets found"
     else: