dtlpy 1.101.8__py3-none-any.whl → 1.102.14__py3-none-any.whl

dtlpy/__version__.py

@@ -1 +1 @@
-version = '1.101.8'
+version = '1.102.14'

dtlpy/entities/app.py

@@ -132,7 +132,7 @@ class App(entities.BaseEntity):
         return self.apps.pause(self)
 
     @staticmethod
-    def _protected_from_json(_json, client_api, project, is_fetched=True):
+    def _protected_from_json(_json, client_api, project=None, is_fetched=True):
         """
         Same as from_json but with try-except to catch if error
 
@@ -193,7 +193,7 @@ class App(entities.BaseEntity):
         return _json
 
     @classmethod
-    def from_json(cls, _json, client_api: ApiClient, project: entities.Project, is_fetched=True):
+    def from_json(cls, _json, client_api: ApiClient, project: entities.Project=None, is_fetched=True):
         app = cls(
             id=_json.get('id', None),
             name=_json.get('name', None),

dtlpy/entities/integration.py

@@ -34,6 +34,8 @@ class IntegrationType(str, Enum):
          - KEY VALUE Integration - for save secrets in the platform
        * - GCP_WORKLOAD_IDENTITY_FEDERATION
          - GCP Workload Identity Federation Integration - for GCP drivers
+       * - PRIVATE_REGISTRY
+         - PRIVATE REGISTRY Integration - for private registry drivers
     """
     S3 = "s3"
     AWS_CROSS_ACCOUNT = 'aws-cross'
@@ -42,7 +44,8 @@ class IntegrationType(str, Enum):
     GCS_CROSS = "gcp-cross"
     AZUREBLOB = "azureblob"
     KEY_VALUE = "key_value"
-    GCP_WORKLOAD_IDENTITY_FEDERATION = "gcp-workload-identity-federation"
+    GCP_WORKLOAD_IDENTITY_FEDERATION = "gcp-workload-identity-federation",
+    PRIVATE_REGISTRY = "private-registry"
 
 
 @attr.s

dtlpy/entities/model.py

@@ -119,7 +119,7 @@ class Model(entities.BaseEntity):
     app = attr.ib(default=None)
 
     @staticmethod
-    def _protected_from_json(_json, client_api, project, package, is_fetched=True):
+    def _protected_from_json(_json, client_api, project=None, package=None, is_fetched=True):
         """
         Same as from_json but with try-except to catch if error
 
@@ -143,7 +143,7 @@ class Model(entities.BaseEntity):
         return status, model
 
     @classmethod
-    def from_json(cls, _json, client_api, project, package, is_fetched=True):
+    def from_json(cls, _json, client_api, project=None, package=None, is_fetched=True):
         """
         Turn platform representation of model into a model entity
 

dtlpy/entities/ontology.py

@@ -185,7 +185,7 @@ class Ontology(entities.BaseEntity):
         self._color_map = values
 
     @staticmethod
-    def _protected_from_json(_json, client_api, recipe, dataset, project, is_fetched=True):
+    def _protected_from_json(_json, client_api, recipe=None, dataset=None, project=None, is_fetched=True):
         """
         Same as from_json but with try-except to catch if error
         :param _json: platform json
@@ -217,7 +217,7 @@ class Ontology(entities.BaseEntity):
         return True
 
     @classmethod
-    def from_json(cls, _json, client_api, recipe, dataset=None, project=None, is_fetched=True):
+    def from_json(cls, _json, client_api, recipe=None, dataset=None, project=None, is_fetched=True):
         """
         Build an Ontology entity object from a json
 

dtlpy/entities/package.py

@@ -124,7 +124,7 @@ class Package(entities.DlEntity):
         self.codebase = entities.ItemCodebase(item_id=item_id)
 
     @staticmethod
-    def _protected_from_json(_json, client_api, project, is_fetched=True):
+    def _protected_from_json(_json, client_api, project=None, is_fetched=True):
         """
         Same as from_json but with try-except to catch if error
 
@@ -144,7 +144,7 @@ class Package(entities.DlEntity):
         return status, package
 
     @classmethod
-    def from_json(cls, _json, client_api, project, is_fetched=True):
+    def from_json(cls, _json, client_api, project=None, is_fetched=True):
         """
         Turn platform representation of package into a package entity
 

dtlpy/entities/pipeline.py

@@ -232,11 +232,12 @@ class Pipeline(entities.BaseEntity):
     updated_by = attr.ib(default=None)
 
     @staticmethod
-    def _protected_from_json(_json, client_api, project, is_fetched=True):
+    def _protected_from_json(_json, client_api, project=None, is_fetched=True):
         """
         Same as from_json but with try-except to catch if error
         :param _json: platform json
         :param client_api: ApiClient entity
+        :param dtlpy.entities.project.Project project: entity
         :param is_fetched: is Entity fetched from Platform
         :return:
         """
@@ -254,13 +255,13 @@ class Pipeline(entities.BaseEntity):
         return status, pipeline
 
     @classmethod
-    def from_json(cls, _json, client_api, project, is_fetched=True):
+    def from_json(cls, _json, client_api, project=None, is_fetched=True):
         """
         Turn platform representation of pipeline into a pipeline entity
 
         :param dict _json: platform representation of package
         :param dl.ApiClient client_api: ApiClient entity
-        :param dtlpy.entities.project.Project project: project entity
+        :param dtlpy.entities.project.Project project: entity
         :param bool is_fetched: is Entity fetched from Platform
         :return: Pipeline entity
         :rtype: dtlpy.entities.pipeline.Pipeline
@@ -308,7 +309,7 @@ class Pipeline(entities.BaseEntity):
         return inst
 
     @classmethod
-    def pipeline_node(self, _json):
+    def pipeline_node(cls, _json):
         node_type = _json.get('type')
         if node_type == 'task':
             return TaskNode.from_json(_json)

dtlpy/entities/pipeline_execution.py

@@ -91,7 +91,7 @@ class PipelineExecution(entities.BaseEntity):
     _repositories = attr.ib(repr=False)
 
     @staticmethod
-    def _protected_from_json(_json, client_api, pipeline, is_fetched=True):
+    def _protected_from_json(_json, client_api, pipeline=None, is_fetched=True):
         """
         Same as from_json but with try-except to catch if error
         :param _json: platform json
@@ -114,7 +114,7 @@ class PipelineExecution(entities.BaseEntity):
         return status, pipeline
 
     @classmethod
-    def from_json(cls, _json, client_api, pipeline, is_fetched=True) -> 'PipelineExecution':
+    def from_json(cls, _json, client_api, pipeline=None, is_fetched=True) -> 'PipelineExecution':
         """
         Turn platform representation of pipeline_execution into a pipeline_execution entity
 

dtlpy/entities/recipe.py

@@ -57,7 +57,7 @@ class Recipe(entities.BaseEntity):
         Build a Recipe entity object from a json
 
         :param dict _json: _json response from host
-        :param dtlpy.entities.dataset.Dataset Dataset: Dataset entity
+        :param dtlpy.entities.dataset.Dataset dataset: Dataset entity
         :param dtlpy.entities.project.Project project: project entity
         :param dl.ApiClient client_api: ApiClient entity
         :param bool is_fetched: is Entity fetched from Platform

dtlpy/entities/task.py

@@ -116,7 +116,7 @@ class Task:
         self._description['content'] = description
 
     @staticmethod
-    def _protected_from_json(_json, client_api, project, dataset):
+    def _protected_from_json(_json, client_api, project=None, dataset=None):
         """
         Same as from_json but with try-except to catch if error
 

dtlpy/entities/trigger.py

@@ -119,7 +119,7 @@ class BaseTrigger(entities.BaseEntity):
         return self.updated_at
 
     @staticmethod
-    def _protected_from_json(_json, client_api, project, service=None):
+    def _protected_from_json(_json, client_api, project=None, service=None):
         """
         Same as from_json but with try-except to catch if error
 
@@ -141,7 +141,7 @@ class BaseTrigger(entities.BaseEntity):
         return status, trigger
 
     @classmethod
-    def from_json(cls, _json, client_api, project, service=None):
+    def from_json(cls, _json, client_api, project=None, service=None):
         """
         Build a trigger entity object from a json
 
@@ -305,7 +305,7 @@ class Trigger(BaseTrigger):
         return _json
 
     @classmethod
-    def from_json(cls, _json, client_api, project, service=None):
+    def from_json(cls, _json, client_api, project=None, service=None):
         """
         Build a trigger entity object from a json
 
@@ -377,7 +377,7 @@ class CronTrigger(BaseTrigger):
         return _json
 
     @classmethod
-    def from_json(cls, _json, client_api, project, service=None):
+    def from_json(cls, _json, client_api, project=None, service=None):
         """
         Build a trigger entity object from a json
 

dtlpy/ml/base_model_adapter.py

@@ -486,7 +486,7 @@ class BaseModelAdapter(utilities.BaseServiceRunner):
         if feature_set is None:
             logger.info('Feature Set not found. creating... ')
             try:
-                self.model_entity.project.feature_sets.get(name=self.model_entity.name)
+                self.model_entity.project.feature_sets.get(feature_set_name=self.model_entity.name)
                 feature_set_name = f"{self.model_entity.name}-{''.join(random.choices(string.ascii_letters + string.digits, k=5))}"
                 logger.warning(f"Feature set with the model name already exists. Creating new feature set with name {feature_set_name}")
             except exceptions.NotFound:

dtlpy/repositories/dpks.py

@@ -196,8 +196,13 @@ class Dpks:
             dpk = dpk_v.items[0]
         return dpk
 
-    def publish(self, dpk: entities.Dpk = None, ignore_max_file_size: bool = False,
-                manifest_filepath='dataloop.json') -> entities.Dpk:
+    def publish(
+            self,
+            dpk: entities.Dpk = None,
+            ignore_max_file_size: bool = False,
+            manifest_filepath='dataloop.json',
+            local_path: str = None
+    ) -> entities.Dpk:
         """
         Upload a dpk entity to the dataloop platform.
 
@@ -207,6 +212,7 @@ class Dpks:
                                         during the packaging of the codebase.
         :param str manifest_filepath: Optional. Path to the manifest file. Can be absolute or relative.
                                     Defaults to 'dataloop.json'
+        :param str local_path: Optional. The path where the dpk files are located.
 
         :return the published dpk
         :rtype dl.entities.Dpk
@@ -241,8 +247,13 @@ class Dpks:
         if self.project and self.project.id != dpk.context['project']:
             logger.warning("the project id that provide different from the dpk project id")
 
+        if local_path is None:
+            if manifest_filepath=='dataloop.json':
+                local_path = os.getcwd()
+            else:
+                local_path = os.path.dirname(manifest_filepath)
         if dpk.codebase is None:
-            dpk.codebase = self.project.codebases.pack(directory=os.getcwd(),
+            dpk.codebase = self.project.codebases.pack(directory=local_path,
                                                        name=dpk.display_name,
                                                        extension='dpk',
                                                        ignore_directories=['artifacts'],

{dtlpy-1.101.8.dist-info → dtlpy-1.102.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dtlpy
-Version: 1.101.8
+Version: 1.102.14
 Summary: SDK and CLI for Dataloop platform
 Home-page: https://github.com/dataloop-ai/dtlpy
 Author: Dataloop Team

{dtlpy-1.101.8.dist-info → dtlpy-1.102.14.dist-info}/RECORD

@@ -1,5 +1,5 @@
 dtlpy/__init__.py,sha256=1Zngp5ftTgWb3r-sc8z98TYpEs6fOB_7snFhsXYQLVg,20899
-dtlpy/__version__.py,sha256=Ozzlk4O9Dg2923lh6fsAFvrBcMuihy5L2HWDivhCqaA,20
+dtlpy/__version__.py,sha256=82fpG2J9U-IcmpwkJvq1am0FxF4AGGDMXe7Jqv7JIi4,21
 dtlpy/exceptions.py,sha256=EQCKs3pwhwZhgMByQN3D3LpWpdxwcKPEEt-bIaDwURM,2871
 dtlpy/new_instance.py,sha256=u_c6JtgqsKCr7TU24-g7_CaST9ghqamMhM4Z0Zxt50w,10121
 dtlpy/assets/__init__.py,sha256=D_hAa6NM8Zoy32sF_9b7m0b7I-BQEyBFg8-9Tg2WOeo,976
@@ -48,7 +48,7 @@ dtlpy/entities/__init__.py,sha256=Sihb652vYCyCWSQfiYTEGQW0EJsHe7BTk6-S5DOsAb0,49
 dtlpy/entities/analytic.py,sha256=5MpYDKPVsZ1MIy20Ju515RWed6P667j4TLxsan2gyNM,11925
 dtlpy/entities/annotation.py,sha256=sqgnONKbNb9gmPLJBH3mYJhFWeKjoiJ8dt57Cz7C6YA,67020
 dtlpy/entities/annotation_collection.py,sha256=CEYSBHhhDkC0VJdHsBSrA6TgdKGMcKeI3tFM40UJwS8,29838
-dtlpy/entities/app.py,sha256=LQ0cDuqeNtO7iihrTq8MBLrFbZiDVs6AFoAi4UQtUZY,6958
+dtlpy/entities/app.py,sha256=dVd87-mP22NWvec5nqA5VjZ8Qk3aJlgUcloIAAOAPUw,6968
 dtlpy/entities/app_module.py,sha256=0UiAbBX1q8iEImi3nY7ySWZZHoRRwu0qUXmyXmgVAc4,3645
 dtlpy/entities/artifact.py,sha256=wtLtBuidOPbnba0ok40JyunCCIBGbAl4bP_ebK39Kk4,5711
 dtlpy/entities/assignment.py,sha256=Dc1QcfVf67GGcmDDi4ubESDuPkSgjXqdqjTBQ31faUM,14722
@@ -66,33 +66,33 @@ dtlpy/entities/feature.py,sha256=9fFjD0W57anOVSAVU55ypxN_WTCsWTG03Wkc3cAAj78,373
 dtlpy/entities/feature_set.py,sha256=niw4MkmrDbD_LWQu1X30uE6U4DCzmFhPTaYeZ6VZDB0,4443
 dtlpy/entities/filters.py,sha256=_A7rLc0yuMw1eW4gSu4-hogQzXbNUheRvFxnTQWntuo,22671
 dtlpy/entities/gis_item.py,sha256=Uk-wMBxwcHsImjz4qOjP-EyZAohbRzN43kMpCaVjCXU,3982
-dtlpy/entities/integration.py,sha256=CA5F1eQCGE_4c_Kry4nWRdeyjHctNnvexcDXg_M5HLU,5734
+dtlpy/entities/integration.py,sha256=j2rbAo__XKOSYJqX9wUYmPtuV4lnbmqEjA0WCrag92k,5876
 dtlpy/entities/item.py,sha256=UnAZ7yLaTu2vkbD2sxysJQNn-ceC9zS3Uf304OvcC4E,31746
 dtlpy/entities/label.py,sha256=ycDYavIgKhz806plIX-64c07_TeHpDa-V7LnfFVe4Rg,3869
 dtlpy/entities/links.py,sha256=FAmEwHtsrqKet3c0UHH9u_gHgG6_OwF1-rl4xK7guME,2516
 dtlpy/entities/message.py,sha256=ApJuaKEqxATpXjNYUjGdYPu3ibQzEMo8-LtJ_4xAcPI,5865
-dtlpy/entities/model.py,sha256=EeJsCePIddavvMVhyNGt6ZIsyteucQoI0cY6byPOi9Y,26878
+dtlpy/entities/model.py,sha256=YwjIi3MxAZoyartTvqx_qhtDKQe6zVsQuwZbYLygMxU,26898
 dtlpy/entities/node.py,sha256=yPPYDLtNMc6vZbbf4FIffY86y7tkaTvYm42Jb7k3Ofk,39617
-dtlpy/entities/ontology.py,sha256=ok4p3sLBc_SS5hs2gZr5-gbblrveM7qSIX4z67QSKeQ,31967
+dtlpy/entities/ontology.py,sha256=N5bsnNUfWNN49WV8lGQL8QRxbKfCZyV-rDyTlGByUrw,31987
 dtlpy/entities/organization.py,sha256=AMkx8hNIIIjnu5pYlNjckMRuKt6H3lnOAqtEynkr7wg,9893
-dtlpy/entities/package.py,sha256=EA5cB3nFBlsbxVK-QroZILjol2bYSVGqCby-mOyJJjQ,26353
+dtlpy/entities/package.py,sha256=QSDePHlp4ik19aUE3dAUC7edh0oUUVjzSmMG867avc4,26363
 dtlpy/entities/package_defaults.py,sha256=wTD7Z7rGYjVy8AcUxTFEnkOkviiJaLVZYvduiUBKNZo,211
 dtlpy/entities/package_function.py,sha256=M42Kvw9A8b6msAkv-wRNAQg_-UC2bejniCjeKDugudc,6314
 dtlpy/entities/package_module.py,sha256=cOkIITATkzzCQpE0sdPiBUisAz8ImlPG2YGZ0K7SypA,5151
 dtlpy/entities/package_slot.py,sha256=XBwCodQe618sQm0bmx46Npo94mEk-zUV7ZX0mDRcsD8,3946
 dtlpy/entities/paged_entities.py,sha256=grNjt2FYg4gKBlVRDkztI1BPOI4JoGeyjvmOW3BnB3k,5927
-dtlpy/entities/pipeline.py,sha256=X9238WbMGfZcXdQVEtkw8twZwl0O4EZB4TxbTSEyPeI,20788
-dtlpy/entities/pipeline_execution.py,sha256=XCXlBAHFYVL2HajE71hK-bPxI4gTwZvg5SKri4BgyRA,9928
+dtlpy/entities/pipeline.py,sha256=JtWGoCUhVszOVkBNK43fbTt446fkND4wH-Y-fN_llww,20851
+dtlpy/entities/pipeline_execution.py,sha256=EQhW4W_G1bIPShYbJSAT--1WNQuvxVQbcQ_MCHIX0KI,9938
 dtlpy/entities/project.py,sha256=ZUx8zA3mr6N145M62R3UDPCCzO1vxfyWO6vjES-bO-g,14653
 dtlpy/entities/prompt_item.py,sha256=d4rqP961PYlJvJJDRXZPI7Z6NdwRXlx_Q0_N0xtZ_B8,19276
-dtlpy/entities/recipe.py,sha256=Q1HtYgind3bEe-vnDZWhw6H-rcIAGhkGHPRWtLIkPSE,11917
+dtlpy/entities/recipe.py,sha256=SX0T7gw-_9Cs2FZyC_htIxQd7CwDwb2zA3SqB37vymM,11917
 dtlpy/entities/reflect_dict.py,sha256=2NaSAL-CO0T0FYRYFQlaSpbsoLT2Q18AqdHgQSLX5Y4,3273
 dtlpy/entities/resource_execution.py,sha256=1HuVV__U4jAUOtOkWlWImnM3Yts8qxMSAkMA9sBhArY,5033
 dtlpy/entities/service.py,sha256=NI4lFC6FqLw4aEGarr2rMptxe3omVfC39C9VAnYYEJA,33733
 dtlpy/entities/setting.py,sha256=uXagJHtcCR3nJYClR_AUGZjz_kx3TejPcUZ8ginHFIA,8561
-dtlpy/entities/task.py,sha256=XHiEqZYFlrDCtmw1MXsysjoBLdIzAk7coMrVk8bNIiE,19534
+dtlpy/entities/task.py,sha256=SL1-6p4jruELkWI-5VXBMn7Imj1xJVaOfAFDa7inH64,19544
 dtlpy/entities/time_series.py,sha256=336jWNckjuSn0G29WJFetB7nBoFAKqs4VH9_IB4m4FE,4017
-dtlpy/entities/trigger.py,sha256=zh3wYUY2-zATh_7ous0Ck87Yojo9r9PAVQrkcESxoko,14266
+dtlpy/entities/trigger.py,sha256=Spf5G3n1PsD3mDntwbAsc-DpEGDlqKgU9ec0Q0HinsQ,14286
 dtlpy/entities/user.py,sha256=hqEzwN6rl1oUTpKOV5eXvw9Z7dtpsiC4TAPSNBmkqcM,3865
 dtlpy/entities/webhook.py,sha256=6R06MgLxabvKySInGlSJmaf0AVmAMe3vKusWhqONRyU,3539
 dtlpy/entities/annotation_definitions/__init__.py,sha256=qZ77hGmCQopPSpiDHYhNWbNKC7nrn10NWNlim9dINmg,666
@@ -148,7 +148,7 @@ dtlpy/miscellaneous/list_print.py,sha256=leEg3RodgYfH5t_0JG8VuM8NiesR8sJLK_mRStt
 dtlpy/miscellaneous/zipping.py,sha256=GMdPhAeHQXeMS5ClaiKWMJWVYQLBLAaJUWxvdYrL4Ro,5337
 dtlpy/ml/__init__.py,sha256=vPkyXpc9kcWWZ_PxyPEOsjKBJdEbowLkZr8FZIb_OBM,799
 dtlpy/ml/base_feature_extractor_adapter.py,sha256=iiEGYAx0Rdn4K46H_FlKrAv3ebTXHSxNVAmio0BxhaI,1178
-dtlpy/ml/base_model_adapter.py,sha256=y7txv5JdTpVCZY6IvYXiQRSz4k3YXhHFEKoDTkcVHz0,50610
+dtlpy/ml/base_model_adapter.py,sha256=FMM1LknEYN69yW12A1f0x6BywdxKteVE_9MuQiryDYo,50622
 dtlpy/ml/metrics.py,sha256=BG2E-1Mvjv2e2No9mIJKVmvzqBvLqytKcw3hA7wVUNc,20037
 dtlpy/ml/predictions_utils.py,sha256=He_84U14oS2Ss7T_-Zj5GDiBZwS-GjMPURUh7u7DjF8,12484
 dtlpy/ml/summary_writer.py,sha256=dehDi8zmGC1sAGyy_3cpSWGXoGQSiQd7bL_Thoo8yIs,2784
@@ -166,7 +166,7 @@ dtlpy/repositories/compositions.py,sha256=H417BvlQAiWr5NH2eANFke6CfEO5o7DSvapYpf
 dtlpy/repositories/computes.py,sha256=EtfE_3JhTdNlSYDPkKXBFkq-DBl4sgQqIm50ajvFdWM,9976
 dtlpy/repositories/datasets.py,sha256=SpG86uToq-E5nVHMwHgWx6VwwwkgfYo8x5vZ0WA3Ouw,56546
 dtlpy/repositories/downloader.py,sha256=rtgGj6jAfXxHZ1oihFoOkK4MUtapFpVMdREKzXKLnu0,44113
-dtlpy/repositories/dpks.py,sha256=mj3QPvfzj_jZAscwIgpKUfa7fLxptc3OJQ_RrSfgYxo,17487
+dtlpy/repositories/dpks.py,sha256=dglvaiSFBvEithhlQ0RAXwzTxoZaICONs-owx3e2nfU,17848
 dtlpy/repositories/drivers.py,sha256=fF0UuHCyBzop8pHfryex23mf0kVFAkqzNdOmwBbaWxY,10204
 dtlpy/repositories/executions.py,sha256=4UoU6bnB3kl5cMuF1eJvDecfZCaB06gKWxPfv6_g1_k,32598
 dtlpy/repositories/feature_sets.py,sha256=UowMDAl_CRefRB5oZzubnsjU_OFgiPPdQXn8q2j4Kuw,9666
@@ -223,19 +223,19 @@ dtlpy/utilities/reports/report.py,sha256=3nEsNnIWmdPEsd21nN8vMMgaZVcPKn9iawKTTeO
 dtlpy/utilities/videos/__init__.py,sha256=SV3w51vfPuGBxaMeNemx6qEMHw_C4lLpWNGXMvdsKSY,734
 dtlpy/utilities/videos/video_player.py,sha256=LCxg0EZ_DeuwcT7U_r7MRC6Q19s0xdFb7x5Gk39PRms,24072
 dtlpy/utilities/videos/videos.py,sha256=Dj916B4TQRIhI7HZVevl3foFrCsPp0eeWwvGbgX3-_A,21875
-dtlpy-1.101.8.data/scripts/dlp,sha256=-F0vSCWuSOOtgERAtsPMPyMmzitjhB7Yeftg_PDlDjw,10
-dtlpy-1.101.8.data/scripts/dlp.bat,sha256=QOvx8Dlx5dUbCTMpwbhOcAIXL1IWmgVRSboQqDhIn3A,37
-dtlpy-1.101.8.data/scripts/dlp.py,sha256=tEokRaDINISXnq8yNx_CBw1qM5uwjYiZoJOYGqWB3RU,4267
+dtlpy-1.102.14.data/scripts/dlp,sha256=-F0vSCWuSOOtgERAtsPMPyMmzitjhB7Yeftg_PDlDjw,10
+dtlpy-1.102.14.data/scripts/dlp.bat,sha256=QOvx8Dlx5dUbCTMpwbhOcAIXL1IWmgVRSboQqDhIn3A,37
+dtlpy-1.102.14.data/scripts/dlp.py,sha256=tEokRaDINISXnq8yNx_CBw1qM5uwjYiZoJOYGqWB3RU,4267
 tests/assets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/assets/models_flow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/assets/models_flow/failedmain.py,sha256=n8F4eu_u7JPrJ1zedbJPvv9e3lHb3ihoErqrBIcseEc,1847
-tests/assets/models_flow/main.py,sha256=xotAjdHpFnIic3Wb-4f7GSg2igtuXZjvRPiYdCTawhA,2064
+tests/assets/models_flow/main.py,sha256=vnDKyVZaae2RFpvwS22Hzi6Dt2LJerH4yQrmKtaT8_g,2123
 tests/assets/models_flow/main_model.py,sha256=Hl_tv7Q6KaRL3yLkpUoLMRqu5-ab1QsUYPL6RPEoamw,2042
 tests/features/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/features/environment.py,sha256=V23cUx_p4VpNk9kc2I0BDZJHO_xcJBFJq8m3JlYCooc,16736
-dtlpy-1.101.8.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-dtlpy-1.101.8.dist-info/METADATA,sha256=EtBmjxKIz7pBOOh-ZOrI_IiLcUhSICMf7oyCNXDKIDI,3019
-dtlpy-1.101.8.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-dtlpy-1.101.8.dist-info/entry_points.txt,sha256=C4PyKthCs_no88HU39eioO68oei64STYXC2ooGZTc4Y,43
-dtlpy-1.101.8.dist-info/top_level.txt,sha256=ZWuLmQGUOtWAdgTf4Fbx884w1o0vBYq9dEc1zLv9Mig,12
-dtlpy-1.101.8.dist-info/RECORD,,
+tests/features/environment.py,sha256=oAO7H7j7Y7czW0t25Gv1KwI2-ofqhZVkbCw9LbZCp4Y,17506
+dtlpy-1.102.14.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+dtlpy-1.102.14.dist-info/METADATA,sha256=8SA6BCGb20bC3aZZwWdE-vrOjyOEEvA7ecIX1aE6fJo,3020
+dtlpy-1.102.14.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+dtlpy-1.102.14.dist-info/entry_points.txt,sha256=C4PyKthCs_no88HU39eioO68oei64STYXC2ooGZTc4Y,43
+dtlpy-1.102.14.dist-info/top_level.txt,sha256=ZWuLmQGUOtWAdgTf4Fbx884w1o0vBYq9dEc1zLv9Mig,12
+dtlpy-1.102.14.dist-info/RECORD,,

tests/assets/models_flow/main.py

@@ -1,3 +1,5 @@
+import time
+
 import dtlpy as dl
 import logging
 import os
@@ -26,10 +28,12 @@ class ModelAdapter(dl.BaseModelAdapter):
 
     def train(self, data_path, output_path, **kwargs):
         logger.info("model training")
+        time.sleep(60)
         print(self.model_entity.id)
 
     def predict(self, batch, **kwargs):
         logger.info("model prediction")
+        time.sleep(30)
         batch_annotations = list()
 
         for img in batch:

tests/features/environment.py

@@ -13,6 +13,7 @@ from behave.formatter.base import StreamOpener
 import sys
 
 import dtlpy as dl
+import shutil
 
 try:
     # for local import
@@ -276,6 +277,11 @@ def after_tag(context, tag):
             use_fixture(reset_setenv, context)
         except Exception:
             logging.exception('Failed to reset env')
+    elif tag == 'restore_json_file':
+        try:
+            use_fixture(restore_json_file, context)
+        except Exception:
+            logging.exception('Failed to restore json file')
     elif tag == 'frozen_dataset':
         pass
     elif 'testrail-C' in tag:
@@ -381,7 +387,8 @@ def delete_pipeline(context):
     while context.to_delete_pipelines_ids:
         pipeline_id = context.to_delete_pipelines_ids.pop(0)
         try:
-            filters = context.dl.Filters(resource=context.dl.FiltersResource.EXECUTION, field='latestStatus.status', values=['created', 'in-progress'], operator='in')
+            filters = context.dl.Filters(resource=context.dl.FiltersResource.EXECUTION, field='latestStatus.status',
+                                         values=['created', 'in-progress'], operator='in')
             filters.add(field='pipeline.id', values=pipeline_id)
             executions = context.dl.executions.list(filters=filters)
             for execution in executions.items:
@@ -494,4 +501,15 @@ def models_delete(context):
         except:
             all_deleted = False
             logging.exception('Failed deleting model: {}'.format(model.id))
-    assert all_deleted
+    assert all_deleted
+
+
+def restore_json_file(context):
+    if not hasattr(context, 'backup_path') or not hasattr(context, 'original_path'):
+        assert False, 'Please make sure to set the original_path and backup_path in the context'
+        # Restore the file from the backup
+    if os.path.exists(context.backup_path):
+        shutil.copy(context.backup_path, context.original_path)
+        os.remove(context.backup_path)  # Clean up the backup
+    else:
+        raise FileNotFoundError(f"Backup file not found for {context.original_path}")