supervisely 6.73.385__py3-none-any.whl → 6.73.386__py3-none-any.whl

supervisely/api/nn/deploy_api.py

@@ -3,6 +3,7 @@
 
 from __future__ import annotations
 
+import os
 import time
 from dataclasses import asdict
 from pathlib import Path
@@ -708,16 +709,12 @@ class DeployApi:
             )
 
         runtime = get_runtime(runtime)
-        experiment_task_id = experiment_info.task_id
-        experiment_task_info = self._api.task.get_info_by_id(experiment_task_id)
-        if experiment_task_info is None:
-            raise ValueError(f"Task with ID '{experiment_task_id}' not found")
-
+        module = None
         module_id = None
         serve_app_name = None
         if with_module:
-            train_module_id = experiment_task_info["meta"]["app"]["moduleId"]
-            module = self.get_serving_app_by_train_app(module_id=train_module_id)
+            framework_name = experiment_info.framework_name
+            module = self.find_serving_app_by_framework(framework_name)
             serve_app_name = module["name"]
             module_id = module["id"]
             logger.debug(f"Serving app detected: '{serve_app_name}'. Module ID: '{module_id}'")
@@ -768,10 +765,15 @@ class DeployApi:
             "runtime": runtime,
         }
 
-        config = experiment_info.model_files.get("config")
-        if config is not None:
-            deploy_params["model_files"]["config"] = f"{experiment_info.artifacts_dir}{config}"
-            logger.debug(f"Config file added: {experiment_info.artifacts_dir}{config}")
+        for file_key, file_path in experiment_info.model_files.items():
+            full_file_path = os.path.join(experiment_info.artifacts_dir, file_path)
+            if not self._api.file.exists(team_id, full_file_path):
+                logger.debug(f"Model file not found: '{full_file_path}'. Trying to find it by checkpoint path.")
+                full_file_path = os.path.join(artifacts_dir, file_path)
+                if not self._api.file.exists(team_id, full_file_path):
+                    raise ValueError(f"Model file not found: '{full_file_path}'. Make sure that the file exists in the artifacts directory.")
+            deploy_params["model_files"][file_key] = full_file_path
+            logger.debug(f"Model file added: {full_file_path}")
         return module_id, serve_app_name, deploy_params
 
     def _set_auto_runtime_by_checkpoint(self, checkpoint_path: str) -> str:

{supervisely-6.73.385.dist-info → supervisely-6.73.386.dist-info}/METADATA

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

{supervisely-6.73.385.dist-info → supervisely-6.73.386.dist-info}/RECORD

@@ -55,7 +55,7 @@ supervisely/api/entity_annotation/figure_api.py,sha256=rmsE3L_JfqN94sLN637pQ0syi
 supervisely/api/entity_annotation/object_api.py,sha256=gbcNvN_KY6G80Me8fHKQgryc2Co7VU_kfFd1GYILZ4E,8875
 supervisely/api/entity_annotation/tag_api.py,sha256=IapvSZmakjdOn0yvqP2tQRY8gkZg0bcvIZBwWRcafrg,18996
 supervisely/api/nn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-supervisely/api/nn/deploy_api.py,sha256=fA5yk3-Q66BL821iu68HpsGWWO2qHXCcWTmrXoHE4gE,37008
+supervisely/api/nn/deploy_api.py,sha256=yxTfi_B8KfKFxv410n4nt4OJiy18lJoZVScXnVCRYgA,37236
 supervisely/api/nn/neural_network_api.py,sha256=VYqpHb6xqCeEUofTNfOGUhvjI_5Di3T26qwVp5Z-0hE,10643
 supervisely/api/pointcloud/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 supervisely/api/pointcloud/pointcloud_annotation_api.py,sha256=xIXqCu0rKYsGt5ezh2EFT2utwsVrr2Xo-MOWUCnbvXc,11259
@@ -1103,9 +1103,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.385.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-supervisely-6.73.385.dist-info/METADATA,sha256=38qHHIiRGEYffwGVbazc2eXQ4_fNNjlxROUqPhBgH-4,35154
-supervisely-6.73.385.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-supervisely-6.73.385.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
-supervisely-6.73.385.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
-supervisely-6.73.385.dist-info/RECORD,,
+supervisely-6.73.386.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+supervisely-6.73.386.dist-info/METADATA,sha256=pAcUu38hzyDhqOlzLArYsO0AzNHXluQ4MbM0UtAm1nk,35154
+supervisely-6.73.386.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+supervisely-6.73.386.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
+supervisely-6.73.386.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
+supervisely-6.73.386.dist-info/RECORD,,