supervisely 6.73.275__py3-none-any.whl → 6.73.276__py3-none-any.whl

supervisely/app/widgets/custom_models_selector/custom_models_selector.py

@@ -25,6 +25,7 @@ from supervisely.app.widgets import (
 )
 from supervisely.io.fs import get_file_name_with_ext
 from supervisely.nn.artifacts.artifacts import TrainInfo
+import supervisely.io.env as sly_env
 
 WEIGHTS_DIR = "weights"
 
@@ -69,18 +70,34 @@ class CustomModelsSelector(Widget):
             # col 1 task
             self._task_id = task_id
             self._task_path = train_info.artifacts_folder
-            task_info = self._api.task.get_info_by_id(task_id)
-            self._task_date_iso = task_info["startedAt"]
-            self._task_date = self._normalize_date()
-            self._task_link = self._create_task_link()
+            try:
+                self._task_info = self._api.task.get_info_by_id(task_id)
+            except:
+                self._task_info = None
+
+            if self._task_info is not None:
+                self._task_date_iso = self._task_info["startedAt"]
+                self._task_date = self._normalize_date()
+                self._task_link = self._create_task_link()
+            else:
+                self._task_date_iso = None
+                self._task_date = None
+                self._task_link = None
             self._config_path = train_info.config_path
 
             # col 2 project
             self._training_project_name = train_info.project_name
-            project_info = self._api.project.get_info_by_name(
-                task_info["workspaceId"], self._training_project_name
+
+            workspace_id = (
+                self._task_info["workspaceId"]
+                if self._task_info
+                else sly_env.workspace_id(raise_not_found=False)
+            )
+            self._training_project_info = (
+                self._api.project.get_info_by_name(workspace_id, self._training_project_name)
+                if workspace_id
+                else None
             )
-            self._training_project_info = project_info
 
             # col 3 checkpoints
             self._checkpoints = train_info.checkpoints
@@ -175,30 +192,36 @@ class CustomModelsSelector(Widget):
                 return ""
 
         def _create_task_widget(self) -> Flexbox:
-            task_widget = Container(
-                [
-                    Text(
-                        f"<i class='zmdi zmdi-folder' style='color: #7f858e'></i> <a href='{self._task_link}'>{self._task_id}</a>",
-                        "text",
-                    ),
-                    Text(
-                        f"<span class='field-description text-muted' style='color: #7f858e'>{self._task_date}</span>",
-                        "text",
-                        font_size=13,
-                    ),
-                ],
-                gap=0,
-            )
+            if self._task_info is not None:
+                task_widget = Container(
+                    [
+                        Text(
+                            f"<i class='zmdi zmdi-folder' style='color: #7f858e'></i> <a href='{self._task_link}'>{self._task_id}</a>",
+                            "text",
+                        ),
+                        Text(
+                            f"<span class='field-description text-muted' style='color: #7f858e'>{self._task_date}</span>",
+                            "text",
+                            font_size=13,
+                        ),
+                    ],
+                    gap=0,
+                )
+            else:
+                task_widget = Text(
+                    f"<span class='field-description text-muted' style='color: #7f858e'>Task was archived (ID: '{self._task_id}')</span>",
+                    "text",
+                )
             return task_widget
 
         def _create_training_project_widget(self) -> Union[ProjectThumbnail, Text]:
-            if self.training_project_info is not None:
+            if self._training_project_info is not None:
                 training_project_widget = ProjectThumbnail(
                     self._training_project_info, remove_margins=True
                 )
             else:
                 training_project_widget = Text(
-                    f"<span class='field-description text-muted' style='color: #7f858e'>Project was deleted</span>",
+                    f"<span class='field-description text-muted' style='color: #7f858e'>Project was archived</span>",
                     "text",
                     font_size=13,
                 )
@@ -209,15 +232,11 @@ class CustomModelsSelector(Widget):
             for checkpoint_info in self._checkpoints:
                 if isinstance(checkpoint_info, dict):
                     checkpoint_selector_items.append(
-                        Select.Item(
-                            value=checkpoint_info["path"], label=checkpoint_info["name"]
-                        )
+                        Select.Item(value=checkpoint_info["path"], label=checkpoint_info["name"])
                     )
                 elif isinstance(checkpoint_info, FileInfo):
                     checkpoint_selector_items.append(
-                        Select.Item(
-                            value=checkpoint_info.path, label=checkpoint_info.name
-                        )
+                        Select.Item(value=checkpoint_info.path, label=checkpoint_info.name)
                     )
 
             checkpoint_selector = Select(items=checkpoint_selector_items)
@@ -282,9 +301,7 @@ class CustomModelsSelector(Widget):
             )
 
             file_api = FileApi(self._api)
-            self._model_path_input = Input(
-                placeholder="Path to model file in Team Files"
-            )
+            self._model_path_input = Input(placeholder="Path to model file in Team Files")
 
             @self._model_path_input.value_changed
             def change_folder(value):
@@ -322,9 +339,7 @@ class CustomModelsSelector(Widget):
 
             self.custom_tab_widgets.hide()
 
-            self.show_custom_checkpoint_path_checkbox = Checkbox(
-                "Use custom checkpoint", False
-            )
+            self.show_custom_checkpoint_path_checkbox = Checkbox("Use custom checkpoint", False)
 
             @self.show_custom_checkpoint_path_checkbox.value_changed
             def show_custom_checkpoint_path_checkbox_changed(is_checked):
@@ -399,9 +414,7 @@ class CustomModelsSelector(Widget):
         self.disable_table()
         super().disable()
 
-    def _generate_table_rows(
-        self, train_infos: List[TrainInfo]
-    ) -> Dict[str, List[ModelRow]]:
+    def _generate_table_rows(self, train_infos: List[TrainInfo]) -> Dict[str, List[ModelRow]]:
         """Method to generate table rows from remote path to training app save directory"""
 
         def process_train_info(train_info):
@@ -414,7 +427,7 @@ class CustomModelsSelector(Widget):
                 )
                 return train_info.task_type, model_row
             except Exception as e:
-                logger.warn(f"Failed to process train info: {train_info}")
+                logger.warning(f"Failed to process train info: {train_info}. Error: {repr(e)}")
                 return None, None
 
         table_rows = defaultdict(list)
@@ -448,8 +461,7 @@ class CustomModelsSelector(Widget):
         if "pose estimation" in task_types:
             sorted_tt.append("pose estimation")
         other_tasks = sorted(
-            set(task_types)
-            - set(["object detection", "instance segmentation", "pose estimation"])
+            set(task_types) - set(["object detection", "instance segmentation", "pose estimation"])
         )
         sorted_tt.extend(other_tasks)
         return sorted_tt
@@ -536,9 +548,7 @@ class CustomModelsSelector(Widget):
 
     def set_custom_checkpoint_task_type(self, task_type: str) -> None:
         if self.use_custom_checkpoint_path():
-            available_task_types = (
-                self.custom_checkpoint_task_type_selector.get_labels()
-            )
+            available_task_types = self.custom_checkpoint_task_type_selector.get_labels()
             if task_type not in available_task_types:
                 raise ValueError(f'"{task_type}" is not available task type')
             self.custom_checkpoint_task_type_selector.set_value(task_type)

{supervisely-6.73.275.dist-info → supervisely-6.73.276.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: supervisely
-Version: 6.73.275
+Version: 6.73.276
 Summary: Supervisely Python SDK.
 Home-page: https://github.com/supervisely/supervisely
 Author: Supervisely

{supervisely-6.73.275.dist-info → supervisely-6.73.276.dist-info}/RECORD

@@ -205,7 +205,7 @@ supervisely/app/widgets/copy_to_clipboard/__init__.py,sha256=47DEQpj8HBSa-_TImW-
 supervisely/app/widgets/copy_to_clipboard/copy_to_clipboard.py,sha256=UlBbXeCBS2_60wtJrMZj3jqPhQJR0wJ3T7vfK4WA588,3283
 supervisely/app/widgets/copy_to_clipboard/template.html,sha256=4gGPot40yVqFkpVVUZGz3DPogVFlZKQ98YM01EUfQDY,492
 supervisely/app/widgets/custom_models_selector/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-supervisely/app/widgets/custom_models_selector/custom_models_selector.py,sha256=yt9rmJR9cHcYorzb04g8yEu5yxrEsRxZVtmlb6-rsMU,20835
+supervisely/app/widgets/custom_models_selector/custom_models_selector.py,sha256=L4FFkTLODWpzrJ09JsCjJA9hNi7ITz9qe1uuRGkf874,21495
 supervisely/app/widgets/custom_models_selector/style.css,sha256=-zPPXHnJvatYj_xVVAb7T8uoSsUTyhm5xCKWkkFQ78E,548
 supervisely/app/widgets/custom_models_selector/template.html,sha256=k7f_Xl6nDUXXwu6IY_RblYni5TbZRRxCBduY5O_SyFs,2908
 supervisely/app/widgets/dataset_thumbnail/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1067,9 +1067,9 @@ supervisely/worker_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 supervisely/worker_proto/worker_api_pb2.py,sha256=VQfi5JRBHs2pFCK1snec3JECgGnua3Xjqw_-b3aFxuM,59142
 supervisely/worker_proto/worker_api_pb2_grpc.py,sha256=3BwQXOaP9qpdi0Dt9EKG--Lm8KGN0C5AgmUfRv77_Jk,28940
 supervisely_lib/__init__.py,sha256=7-3QnN8Zf0wj8NCr2oJmqoQWMKKPKTECvjH9pd2S5vY,159
-supervisely-6.73.275.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-supervisely-6.73.275.dist-info/METADATA,sha256=zkN67LRGghMe6rNZp9mRGpe8gal1OLLXh49AyenpU-I,33573
-supervisely-6.73.275.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-supervisely-6.73.275.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
-supervisely-6.73.275.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
-supervisely-6.73.275.dist-info/RECORD,,
+supervisely-6.73.276.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+supervisely-6.73.276.dist-info/METADATA,sha256=IEQlHjL5OYrRUVXISxSa5jZEaxfStqtTGbn40HHu810,33573
+supervisely-6.73.276.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+supervisely-6.73.276.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
+supervisely-6.73.276.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
+supervisely-6.73.276.dist-info/RECORD,,