From 7cb9f69b841e4d9619a229ff492fc3f16e684d47 Mon Sep 17 00:00:00 2001
From: Almaz <79905215+almazgimaev@users.noreply.github.com>
Date: Tue, 10 Dec 2024 22:49:35 +0100
Subject: [PATCH] Upgrade SDK to 6.73.250 to fix project downloading errors
 (#78)

---
 serve/config.json          | 2 +-
 serve/dev_requirements.txt | 2 +-
 train/config.json          | 2 +-
 train/dev_requirements.txt | 2 +-
 train/src/main.py          | 5 ++++-
 5 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/serve/config.json b/serve/config.json
index 581eb86..fe55be3 100644
--- a/serve/config.json
+++ b/serve/config.json
@@ -20,7 +20,7 @@
     "deployed_nn"
   ],
   "community_agent": false,
-  "docker_image": "supervisely/yolov8:1.0.39",
+  "docker_image": "supervisely/yolov8:1.0.40",
   "instance_version": "6.12.12",
   "entrypoint": "python3 -m uvicorn src.main:m.app --app-dir ./serve --host 0.0.0.0 --port 8000 --ws websockets",
   "port": 8000,
diff --git a/serve/dev_requirements.txt b/serve/dev_requirements.txt
index a79fa63..3dc9c67 100644
--- a/serve/dev_requirements.txt
+++ b/serve/dev_requirements.txt
@@ -1,6 +1,6 @@
 # git+https://github.com/supervisely/supervisely.git@some-test-branch
 
-supervisely[model-benchmark]==6.73.232
+supervisely[model-benchmark]==6.73.250
 ultralytics==8.2.26
 --extra-index-url https://download.pytorch.org/whl/cu113
 torch==1.10.1+cu113
diff --git a/train/config.json b/train/config.json
index 5074f09..d1aab14 100644
--- a/train/config.json
+++ b/train/config.json
@@ -11,7 +11,7 @@
     "train"
   ],
   "description": "Dashboard to configure, start and monitor YOLOv8 | v9 | v10 | v11 training",
-  "docker_image": "supervisely/yolov8:1.0.39",
+  "docker_image": "supervisely/yolov8:1.0.40",
   "instance_version": "6.12.12",
   "entrypoint": "python3 -m uvicorn src.main:app --app-dir ./train --host 0.0.0.0 --port 8000 --ws websockets",
   "task_location": "workspace_tasks",
diff --git a/train/dev_requirements.txt b/train/dev_requirements.txt
index 5b8fa35..3eaa106 100644
--- a/train/dev_requirements.txt
+++ b/train/dev_requirements.txt
@@ -1 +1 @@
-supervisely[model-benchmark]==6.73.239
+supervisely[model-benchmark]==6.73.250
diff --git a/train/src/main.py b/train/src/main.py
index a581d67..c2e72a0 100644
--- a/train/src/main.py
+++ b/train/src/main.py
@@ -230,7 +230,10 @@ def update_globals(new_dataset_ids):
 ### 1. Dataset selection
 # dataset_selector = SelectDataset(project_id=project_id, multiselect=True, select_all_datasets=True)
 dataset_selector = SelectDatasetTree(
-    project_id=project_id, multiselect=True, select_all_datasets=True
+    project_id=project_id,
+    multiselect=True,
+    select_all_datasets=True,
+    allowed_project_types=[sly.ProjectType.IMAGES],
 )
 use_cache_text = Text(
     "Use cached data stored on the agent to optimize project download"