zenml-nightly 0.66.0.dev20240910__py3-none-any.whl → 0.66.0.dev20240918__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.66.0.dev20240910
+0.66.0.dev20240918

zenml/cli/service_accounts.py

@@ -35,6 +35,7 @@ def _create_api_key(
     name: str,
     description: Optional[str],
     set_key: bool = False,
+    output_file: Optional[str] = None,
 ) -> None:
     """Create an API key.
 
@@ -44,6 +45,7 @@ def _create_api_key(
         name: Name of the API key
         description: The API key description.
         set_key: Configure the local client with the generated key.
+        output_file: Output file to write the API key to.
     """
     client = Client()
     zen_store = client.zen_store
@@ -74,13 +76,19 @@ def _create_api_key(
             )
             return
 
-    cli_utils.declare(
-        f"The API key value is: '{api_key.key}'\nPlease store it safely as "
-        "it will not be shown again.\nTo configure a ZenML client to use "
-        "this API key, run:\n\n"
-        f"zenml connect --url {zen_store.config.url} --api-key \\\n"
-        f"    '{api_key.key}'\n"
-    )
+    if output_file and api_key.key:
+        with open(output_file, "w") as f:
+            f.write(api_key.key)
+
+        cli_utils.declare(f"Wrote API key value to {output_file}")
+    else:
+        cli_utils.declare(
+            f"The API key value is: '{api_key.key}'\nPlease store it safely as "
+            "it will not be shown again.\nTo configure a ZenML client to use "
+            "this API key, run:\n\n"
+            f"zenml connect --url {zen_store.config.url} --api-key \\\n"
+            f"    '{api_key.key}'\n"
+        )
 
 
 @cli.group(cls=TagGroup, tag=CliCategories.IDENTITY_AND_SECURITY)
@@ -111,11 +119,18 @@ def service_account() -> None:
     help=("Configure the local client to use the generated API key."),
     is_flag=True,
 )
+@click.option(
+    "--output-file",
+    type=str,
+    required=False,
+    help="File to write the API key to.",
+)
 def create_service_account(
     service_account_name: str,
     description: str = "",
     create_api_key: bool = True,
     set_api_key: bool = False,
+    output_file: Optional[str] = None,
 ) -> None:
     """Create a new service account.
 
@@ -124,6 +139,7 @@ def create_service_account(
         description: The API key description.
         create_api_key: Create an API key for the service account.
         set_api_key: Configure the local client to use the generated API key.
+        output_file: Output file to write the API key to.
     """
     client = Client()
     try:
@@ -142,6 +158,7 @@ def create_service_account(
             name="default",
             description="Default API key.",
             set_key=set_api_key,
+            output_file=output_file,
         )
 
 
@@ -302,12 +319,19 @@ def api_key(
     is_flag=True,
     help="Configure the local client with the generated key.",
 )
+@click.option(
+    "--output-file",
+    type=str,
+    required=False,
+    help="File to write the API key to.",
+)
 @click.pass_obj
 def create_api_key(
     service_account_name_or_id: str,
     name: str,
     description: Optional[str],
     set_key: bool = False,
+    output_file: Optional[str] = None,
 ) -> None:
     """Create an API key.
 
@@ -317,12 +341,14 @@ def create_api_key(
         name: Name of the API key
         description: The API key description.
         set_key: Configure the local client with the generated key.
+        output_file: Output file to write the API key to.
     """
     _create_api_key(
         service_account_name_or_id=service_account_name_or_id,
         name=name,
         description=description,
         set_key=set_key,
+        output_file=output_file,
     )
 
 
@@ -450,12 +476,19 @@ def update_api_key(
     is_flag=True,
     help="Configure the local client with the generated key.",
 )
+@click.option(
+    "--output-file",
+    type=str,
+    required=False,
+    help="File to write the API key to.",
+)
 @click.pass_obj
 def rotate_api_key(
     service_account_name_or_id: str,
     name_or_id: str,
     retain: int = 0,
     set_key: bool = False,
+    output_file: Optional[str] = None,
 ) -> None:
     """Rotate an API key.
 
@@ -466,6 +499,7 @@ def rotate_api_key(
         retain: Number of minutes for which the previous key is still valid
             after it has been rotated.
         set_key: Configure the local client with the newly generated key.
+        output_file: Output file to write the API key to.
     """
     client = Client()
     zen_store = client.zen_store
@@ -499,13 +533,19 @@ def rotate_api_key(
             )
             return
 
-    cli_utils.declare(
-        f"The new API key value is: '{api_key.key}'\nPlease store it "
-        "safely as it will not be shown again.\nTo configure a ZenML "
-        "client to use this API key, run:\n\n"
-        f"zenml connect --url {zen_store.config.url} --api-key \\\n"
-        f"    '{api_key.key}'\n"
-    )
+    if output_file and api_key.key:
+        with open(output_file, "w") as f:
+            f.write(api_key.key)
+
+        cli_utils.declare(f"Wrote API key value to {output_file}")
+    else:
+        cli_utils.declare(
+            f"The new API key value is: '{api_key.key}'\nPlease store it "
+            "safely as it will not be shown again.\nTo configure a ZenML "
+            "client to use this API key, run:\n\n"
+            f"zenml connect --url {zen_store.config.url} --api-key \\\n"
+            f"    '{api_key.key}'\n"
+        )
 
 
 @api_key.command("delete")

zenml/client.py

@@ -3748,6 +3748,7 @@ class Client(metaclass=ClientMetaClass):
         end_time: Optional[Union[datetime, str]] = None,
         num_steps: Optional[Union[int, str]] = None,
         unlisted: Optional[bool] = None,
+        templatable: Optional[bool] = None,
         tag: Optional[str] = None,
         hydrate: bool = False,
     ) -> Page[PipelineRunResponse]:
@@ -3778,6 +3779,7 @@ class Client(metaclass=ClientMetaClass):
             end_time: The end_time for the pipeline run
             num_steps: The number of steps for the pipeline run
             unlisted: If the runs should be unlisted or not.
+            templatable: If the runs should be templatable or not.
             tag: Tag to filter by.
             hydrate: Flag deciding whether to hydrate the output model(s)
                 by including metadata fields in the response.
@@ -3811,6 +3813,7 @@ class Client(metaclass=ClientMetaClass):
             num_steps=num_steps,
             tag=tag,
             unlisted=unlisted,
+            templatable=templatable,
         )
         runs_filter_model.set_scope_workspace(self.active_workspace.id)
         return self.zen_store.list_runs(

zenml/models/v2/core/pipeline_run.py

@@ -235,6 +235,9 @@ class PipelineRunResponseMetadata(WorkspaceScopedResponseMetadata):
         default=None,
         description="Template used for the pipeline run.",
     )
+    is_templatable: bool = Field(
+        description="Whether a template can be created from this run.",
+    )
 
 
 class PipelineRunResponseResources(WorkspaceScopedResponseResources):
@@ -477,6 +480,15 @@ class PipelineRunResponse(
         """
         return self.get_metadata().template_id
 
+    @property
+    def is_templatable(self) -> bool:
+        """The `is_templatable` property.
+
+        Returns:
+            the value of the property.
+        """
+        return self.get_metadata().is_templatable
+
     @property
     def model_version(self) -> Optional[ModelVersionResponse]:
         """The `model_version` property.
@@ -511,6 +523,7 @@ class PipelineRunFilter(WorkspaceScopedTaggableFilter):
         "stack_id",
         "template_id",
         "pipeline_name",
+        "templatable",
     ]
     name: Optional[str] = Field(
         default=None,
@@ -584,6 +597,7 @@ class PipelineRunFilter(WorkspaceScopedTaggableFilter):
         union_mode="left_to_right",
     )
     unlisted: Optional[bool] = None
+    templatable: Optional[bool] = None
 
     def get_custom_filters(
         self,
@@ -595,7 +609,7 @@ class PipelineRunFilter(WorkspaceScopedTaggableFilter):
         """
         custom_filters = super().get_custom_filters()
 
-        from sqlmodel import and_
+        from sqlmodel import and_, col, or_
 
         from zenml.zen_stores.schemas import (
             CodeReferenceSchema,
@@ -668,4 +682,40 @@ class PipelineRunFilter(WorkspaceScopedTaggableFilter):
             )
             custom_filters.append(run_template_filter)
 
+        if self.templatable is not None:
+            if self.templatable is True:
+                templatable_filter = and_(
+                    # The following condition is not perfect as it does not
+                    # consider stacks with custom flavor components or local
+                    # components, but the best we can do currently with our
+                    # table columns.
+                    PipelineRunSchema.deployment_id
+                    == PipelineDeploymentSchema.id,
+                    PipelineDeploymentSchema.build_id
+                    == PipelineBuildSchema.id,
+                    col(PipelineBuildSchema.is_local).is_(False),
+                    col(PipelineBuildSchema.stack_id).is_not(None),
+                )
+            else:
+                templatable_filter = or_(
+                    col(PipelineRunSchema.deployment_id).is_(None),
+                    and_(
+                        PipelineRunSchema.deployment_id
+                        == PipelineDeploymentSchema.id,
+                        col(PipelineDeploymentSchema.build_id).is_(None),
+                    ),
+                    and_(
+                        PipelineRunSchema.deployment_id
+                        == PipelineDeploymentSchema.id,
+                        PipelineDeploymentSchema.build_id
+                        == PipelineBuildSchema.id,
+                        or_(
+                            col(PipelineBuildSchema.is_local).is_(True),
+                            col(PipelineBuildSchema.stack_id).is_(None),
+                        ),
+                    ),
+                )
+
+            custom_filters.append(templatable_filter)
+
         return custom_filters

zenml/service_connectors/docker_service_connector.py

@@ -261,8 +261,18 @@ class DockerServiceConnector(ServiceConnector):
         """
         assert self.resource_id is not None
 
+        # The reason we need to configure the local client here instead of
+        # using the `docker_client.login(...)` method is the following:
+        # When calling the login method on the python client, it stores the
+        # credentials in memory, but doesn't actually use them in future calls
+        # to push/pull images in case a credential store or credential helper is
+        # configured. If the cred store/helper contains invalid/expired
+        # credentials for the registry, these calls will fail.
+        # This solution is not ideal as we're now replacing potentially
+        # long-lived credentials in the cred store with our short lived ones,
+        # but the best we can do without modifying the docker library.
+        self._configure_local_client()
         docker_client = docker_utils._try_get_docker_client_from_env()
-
         self._authorize_client(docker_client, self.resource_id)
 
         return docker_client

zenml/stack/stack.py

@@ -751,21 +751,6 @@ class Stack:
                 updated=datetime.utcnow(),
             )
 
-            logger.warning(
-                "The stack `%s` contains components that require building "
-                "Docker images. Older versions of ZenML always built these "
-                "images locally, but since version 0.32.0 this behavior can be "
-                "configured using the `image_builder` stack component. This "
-                "stack will temporarily default to a local image builder that "
-                "mirrors the previous behavior, but this will be removed in "
-                "future versions of ZenML. Please add an image builder to this "
-                "stack:\n"
-                "`zenml image-builder register <NAME> ...\n"
-                "zenml stack update %s -i <NAME>`",
-                self.name,
-                self.id,
-            )
-
             self._image_builder = image_builder
 
     def prepare_pipeline_deployment(

zenml/stack/stack_component.py

@@ -421,23 +421,56 @@ class StackComponent:
         else:
             user_id = None
 
-        return flavor.implementation_class(
-            user=user_id,
-            workspace=component_model.workspace.id,
-            name=component_model.name,
-            id=component_model.id,
-            config=configuration,
-            labels=component_model.labels,
-            flavor=component_model.flavor,
-            type=component_model.type,
-            created=component_model.created,
-            updated=component_model.updated,
-            connector_requirements=flavor.service_connector_requirements,
-            connector=component_model.connector.id
-            if component_model.connector
-            else None,
-            connector_resource_id=component_model.connector_resource_id,
-        )
+        try:
+            return flavor.implementation_class(
+                user=user_id,
+                workspace=component_model.workspace.id,
+                name=component_model.name,
+                id=component_model.id,
+                config=configuration,
+                labels=component_model.labels,
+                flavor=component_model.flavor,
+                type=component_model.type,
+                created=component_model.created,
+                updated=component_model.updated,
+                connector_requirements=flavor.service_connector_requirements,
+                connector=component_model.connector.id
+                if component_model.connector
+                else None,
+                connector_resource_id=component_model.connector_resource_id,
+            )
+        except ImportError as e:
+            from zenml.integrations.registry import integration_registry
+
+            integration_requirements = " ".join(
+                integration_registry.select_integration_requirements(
+                    flavor_model.integration
+                )
+            )
+
+            if integration_registry.is_installed(flavor_model.integration):
+                raise ImportError(
+                    f"{e}\n\n"
+                    f"Something went wrong while trying to import from the "
+                    f"`{flavor_model.integration}` integration. Please make "
+                    "sure that all its requirements are installed properly by "
+                    "reinstalling the integration either through our CLI: "
+                    f"`zenml integration install {flavor_model.integration} "
+                    "-y` or by manually installing its requirements: "
+                    f"`pip install {integration_requirements}`. If the error"
+                    "persists, please contact the ZenML team."
+                ) from e
+            else:
+                raise ImportError(
+                    f"{e}\n\n"
+                    f"The `{flavor_model.integration}` integration that you "
+                    "are trying to use is not installed in your current "
+                    "environment. Please make sure that it is installed by "
+                    "either using our CLI: `zenml integration install "
+                    f"{flavor_model.integration}` or by manually installing "
+                    f"its requirements: `pip install "
+                    f"{integration_requirements}`"
+                ) from e
 
     @property
     def config(self) -> StackComponentConfig:

zenml/steps/base_step.py

@@ -592,16 +592,34 @@ class BaseStep(metaclass=BaseStepMeta):
         from zenml.new.pipelines.pipeline import Pipeline
 
         if not Pipeline.ACTIVE_PIPELINE:
-            # The step is being called outside the context of a pipeline, either
-            # run the step function or run it as a single step pipeline on the
-            # active stack
+            from zenml import constants, get_step_context
+
+            # If the environment variable was set to explicitly not run on the
+            # stack, we do that.
             run_without_stack = handle_bool_env_var(
                 ENV_ZENML_RUN_SINGLE_STEPS_WITHOUT_STACK, default=False
             )
             if run_without_stack:
                 return self.call_entrypoint(*args, **kwargs)
+
+            try:
+                get_step_context()
+            except RuntimeError:
+                pass
             else:
-                return run_as_single_step_pipeline(self, *args, **kwargs)
+                # We're currently inside the execution of a different step
+                # -> We don't want to launch another single step pipeline here,
+                # but instead just call the step function
+                return self.call_entrypoint(*args, **kwargs)
+
+            if constants.SHOULD_PREVENT_PIPELINE_EXECUTION:
+                logger.info(
+                    "Preventing execution of step '%s'.",
+                    self.name,
+                )
+                return
+
+            return run_as_single_step_pipeline(self, *args, **kwargs)
 
         (
             input_artifacts,

zenml/zen_stores/schemas/pipeline_run_schemas.py

@@ -328,6 +328,15 @@ class PipelineRunSchema(NamedSchema, table=True):
         )
         metadata = None
         if include_metadata:
+            is_templatable = False
+            if (
+                self.deployment
+                and self.deployment.build
+                and not self.deployment.build.is_local
+                and self.deployment.build.stack
+            ):
+                is_templatable = True
+
             steps = {step.name: step.to_model() for step in self.step_runs}
 
             metadata = PipelineRunResponseMetadata(
@@ -346,6 +355,7 @@ class PipelineRunSchema(NamedSchema, table=True):
                 template_id=self.deployment.template_id
                 if self.deployment
                 else None,
+                is_templatable=is_templatable,
             )
 
         resources = None

zenml/zen_stores/schemas/run_template_schemas.py

@@ -177,6 +177,7 @@ class RunTemplateSchema(BaseSchema, table=True):
         if (
             self.source_deployment
             and self.source_deployment.build
+            and not self.source_deployment.build.is_local
             and self.source_deployment.build.stack
         ):
             runnable = True

zenml/zen_stores/sql_zen_store.py

@@ -974,6 +974,7 @@ class SqlZenStore(BaseZenStore):
             RuntimeError: if the schema does not have a `to_model` method.
         """
         query = filter_model.apply_filter(query=query, table=table)
+        query = query.distinct()
 
         # Get the total amount of items in the database for a given query
         custom_fetch_result: Optional[Sequence[Any]] = None
@@ -7218,6 +7219,8 @@ class SqlZenStore(BaseZenStore):
         )
 
     # ----------------------------- Stacks -----------------------------
+
+    @track_decorator(AnalyticsEvent.REGISTERED_STACK)
     def create_stack(self, stack: StackRequest) -> StackResponse:
         """Register a full stack.
 

{zenml_nightly-0.66.0.dev20240910.dist-info → zenml_nightly-0.66.0.dev20240918.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zenml-nightly
-Version: 0.66.0.dev20240910
+Version: 0.66.0.dev20240918
 Summary: ZenML: Write production-ready ML code.
 Home-page: https://zenml.io
 License: Apache-2.0

{zenml_nightly-0.66.0.dev20240910.dist-info → zenml_nightly-0.66.0.dev20240918.dist-info}/RECORD

@@ -6,7 +6,7 @@ RELEASE_NOTES.md,sha256=GjuHlUd-5AcepXC8ETMkIRMxKVQtBekiuY_CW0oP4XQ,365106
 ROADMAP.md,sha256=hiLSmr16BH8Dfx7SaQM4JcXCGCVl6mFZPFAwJeDTrJU,407
 SECURITY.md,sha256=9DepA8y03yvCZLHEfcXLTDH4lUyKHquAdukBsccNN7c,682
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=IIehTiT1H_PJFlSTpiWfIJ2bnVYUB7OQ3S6LYuDUvKI,19
+zenml/VERSION,sha256=EFLZgxc91tLiAqNwF0fItVSwNRCUEwoiosAP0HS1FJI,19
 zenml/__init__.py,sha256=X_JnjAYTBG2ZYgcM2_FDPssygdj7z7VTAmU9HEi17p0,2306
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -51,7 +51,7 @@ zenml/cli/pipeline.py,sha256=8UO3ZjKZbaU3tiRLBsN-N3kK_i_bD-8Qbq_YpR8mstw,17442
 zenml/cli/secret.py,sha256=2WxDYg6iodnAVSbVJQvQEYWEgcy0oytcSBVnnyN0JOw,20135
 zenml/cli/served_model.py,sha256=lIPd6kOS9kfWFOsd3DSILtiYomINpmXrdpyP5vtQlj8,14604
 zenml/cli/server.py,sha256=N2cvu9C_z8CzwGu3a7P6fE2X8Se_rHEx30iq9nYMy9o,28682
-zenml/cli/service_accounts.py,sha256=MO8pVVjppdvIULFxpvQKlhfEDmRqvlH8aIbPTdNSHXU,16489
+zenml/cli/service_accounts.py,sha256=V6Oeomcs0tg-MIOUbYABqTcO4PsGCdM2WXJH576iqKM,17723
 zenml/cli/service_connectors.py,sha256=v6o-FpTEnDXYcQXczBELwfWPCcYBi9z7b1GC90Fvp30,74397
 zenml/cli/stack.py,sha256=vT-7Kac_Cf2g9XoIzwrxdQYgtkpuv_d-BpDvL-p9XUk,82862
 zenml/cli/stack_components.py,sha256=BxWsqeu5e2DtP5TMjzE98d8Lqfwe5o45TC2awjil5wY,74144
@@ -63,7 +63,7 @@ zenml/cli/utils.py,sha256=whFzoFBxlDlldLahW7YzZc8HWcb7vGk0Jr5O5VlGcLk,91136
 zenml/cli/version.py,sha256=nm1iSU_1V6-MUwpMKeXcwFhLYGUMLswvQL67cEuCpxA,3635
 zenml/cli/web_login.py,sha256=yqJ4VRcnyz-QhWIyR2wfGpSVo_3pe3vOcRRYdluJsgI,7647
 zenml/cli/workspace.py,sha256=bp02aXou574ToWPD8OAIB_cg3mvpE011H8aMKegT-nU,2970
-zenml/client.py,sha256=HrwvGhrcNicL8SLDADinx4dZAJ753c8SCm19Pl4W8A4,279610
+zenml/client.py,sha256=IZw7AGeqI84zjIE3Bv8r31NvxQY7_0KmDGsPcoAH468,279758
 zenml/client_lazy_loader.py,sha256=O3PK5yB3dfqdKINfOs_LNsmnOVTvKsUl80r3JsvKldQ,6570
 zenml/code_repositories/__init__.py,sha256=W5bDfzAG8OXIKZSV1L-VHuzMcSCYa9qzTdPb3jqfyYw,920
 zenml/code_repositories/base_code_repository.py,sha256=_DbxIBxvJlN0PFhST0vlTIQ26Q6V3Nar0kYdeGaJrk8,4386
@@ -618,7 +618,7 @@ zenml/models/v2/core/model_version_pipeline_run.py,sha256=ghxCdCckYyzOwiJHhB0LU0
 zenml/models/v2/core/pipeline.py,sha256=ZaKZ-Y0bmB6mokckLRpMXgR-UMAIwsa9NSRrdKoq6_k,8368
 zenml/models/v2/core/pipeline_build.py,sha256=-Y-9V1pcHq-jI2VPEyT-Chw5hGQMljvbcDF8s2-wiko,14875
 zenml/models/v2/core/pipeline_deployment.py,sha256=J10yhxqzSPXqTQZZPIN1zhe-yrZ5cEVvilkXBRiZwjg,12451
-zenml/models/v2/core/pipeline_run.py,sha256=VqxC8rI1OXbpuimmkm-F4dpyHrzQGke-hq-f8bhGwy0,21137
+zenml/models/v2/core/pipeline_run.py,sha256=qMwbqapkJFhBdLABtPZAjzrczmNOorNjTFdxkkpsuAE,23192
 zenml/models/v2/core/run_metadata.py,sha256=u7c3LtZikb6EY6uwhXuqSIcaOgb0jDfGU8tQt3Ccerk,7863
 zenml/models/v2/core/run_template.py,sha256=baa6q0z40n6sa5XgmTXxS2gJmpq-kFvu7jnXMjWu47g,11728
 zenml/models/v2/core/schedule.py,sha256=QE6_2KFmamGMClS7-sE43hyNBfkVnml6rDqvmRmyiB8,10215
@@ -692,7 +692,7 @@ zenml/secret/schemas/azure_secret_schema.py,sha256=3WP-DRgqoHJVsJit8fQcxjd1KYX2E
 zenml/secret/schemas/basic_auth_secret_schema.py,sha256=q-4dDtycHyWkIdZGd-qnIp4n8cxZO13qT7w_sTA4hWY,1021
 zenml/secret/schemas/gcp_secret_schema.py,sha256=tDSxPXlfh4BOSICZjfJjUukfzEyBEgxpQzk4y-OV1AE,1677
 zenml/service_connectors/__init__.py,sha256=gIBAVUPBZtdO6JtEBCixy9YG4wZRA1mnaxaGAxi3T1Q,645
-zenml/service_connectors/docker_service_connector.py,sha256=vplyZNUxZhAyzXRdtAiSGMFZT87h9Wq8izE0HWh7OSQ,12441
+zenml/service_connectors/docker_service_connector.py,sha256=_6WPqYCcSUf8JPakipbkJRvaN2ghhY7zCr38WCHG0SI,13218
 zenml/service_connectors/service_connector.py,sha256=HW8dTE826ZSD1vWQlqQRm8XYq08v7tDZQ8ORnJ5rTcE,55351
 zenml/service_connectors/service_connector_registry.py,sha256=lP0s0k7RIp9WZEW36Q0ygHD5urzF-wUwh41rbXWS5ss,9186
 zenml/service_connectors/service_connector_utils.py,sha256=h3wfVSPO0sQnARVRnih3l8P7LwkCmAgijThuHp2Z53E,18026
@@ -716,8 +716,8 @@ zenml/stack/__init__.py,sha256=vfHzaoRhPtS-XSlM8Vx1noJZDHF1Pj6LDz2knpn_QBg,1236
 zenml/stack/authentication_mixin.py,sha256=sg7GkpB-Ao9Gsa7Po0jxMn-_mVYUB42etmspZ6dk8cI,3982
 zenml/stack/flavor.py,sha256=rut-jFQFB6OGm25RRyMtYlJ3mm_6wrHYVI54Hmf9te4,9726
 zenml/stack/flavor_registry.py,sha256=IL0fRrxxQJ9YkCYCeADP7nwWEQo4XBElJY4owMjKGbQ,6108
-zenml/stack/stack.py,sha256=5pgR30_Xgj4Qi09zKYyleuaWwMhe0zvi4k7yF1r8_BM,38021
-zenml/stack/stack_component.py,sha256=DpWsgaqvPv9bJhACdMklL-Mjo453TjzQL4JVMzI3PQU,29887
+zenml/stack/stack.py,sha256=JIRSNX1fRCubXGPCg-qjeVtlT9fDeAX-xiToDa9CeVk,37235
+zenml/stack/stack_component.py,sha256=9sgAu2EGZ2yiEn_D0dq_kmytFmlt8spJ3iG0GlHIFVU,31634
 zenml/stack/stack_validator.py,sha256=hWbvvGIeWLj6NwSsF4GCc6RAxAWvxHXTcBZL9nJvcak,3111
 zenml/stack/utils.py,sha256=dcJsbLZf-hJYM9Ut8BfNzrWygmz6shHegqHPRVLoLWE,5840
 zenml/stack_deployments/__init__.py,sha256=-7593cQ_ZgRn774Ol-8AKXXQquIU4DSiaThVEr6TfWM,644
@@ -731,7 +731,7 @@ zenml/step_operators/base_step_operator.py,sha256=ZRnY6lcEHY8xZskrKKdPhgKg2BlEoh
 zenml/step_operators/step_operator_entrypoint_configuration.py,sha256=vG0Z9D2C72PYOZ_P3eXYTmm2TvL-dS_kHhSMq-coAKE,3690
 zenml/steps/__init__.py,sha256=IvxepI3tYp6LiVHFJnet9xZ7yFlH6NHCAI69SvQsfaI,1682
 zenml/steps/base_parameters.py,sha256=rKnALd0lMv0jOX8E9TwHJkeP3H-ijEFXY0T5dMCu9EQ,756
-zenml/steps/base_step.py,sha256=RsSIETBgBYk3phwTzN407iJy-Lp8Mtdzni8zNP4g7Sg,51852
+zenml/steps/base_step.py,sha256=sQbGnPt_1-upcYAGPqqKW0xrULNZZc9NR_7fu2y5qQk,52434
 zenml/steps/entrypoint_function_utils.py,sha256=25LxlZ8E-3YoWEdAUFK4UZ04hbp4GNMqpKwD2O-Rp_0,12484
 zenml/steps/external_artifact.py,sha256=63ngwlSJvR8GoM93ttz1Mp6BuAg7xH-QZAsBvW3M80Q,1109
 zenml/steps/step_decorator.py,sha256=_rV5MGMziqNiOoxupZUy055o-a0b8Y4RtLuB6nLyYyg,8188
@@ -1370,10 +1370,10 @@ zenml/zen_stores/schemas/logs_schemas.py,sha256=qv6fs3JiVgzlmTXJqb_gG5NsU5q_50e0
 zenml/zen_stores/schemas/model_schemas.py,sha256=wLf1SeJPPzyq7Bm7oTTBdVHfZbOAWw027aKn_TJZ_JI,25559
 zenml/zen_stores/schemas/pipeline_build_schemas.py,sha256=QK3sVPYhumUx_x-H3YZ-RfnsxGhKfkMGdP5FC9WrQgU,5718
 zenml/zen_stores/schemas/pipeline_deployment_schemas.py,sha256=ihJswWXvaXilyNS8RnOfnZM1Yi1B9t7PN8t7f1vOoho,10177
-zenml/zen_stores/schemas/pipeline_run_schemas.py,sha256=LbGX3Srz-zbhnCY4th9SVSlNoF3e5ss9WP4E1p5ODWw,15300
+zenml/zen_stores/schemas/pipeline_run_schemas.py,sha256=PP_r3HI9oWwV0v0lewUyf7LUUZiXlZ1FR53Y2NrWbFQ,15630
 zenml/zen_stores/schemas/pipeline_schemas.py,sha256=HjQfuH3WDpkf_gXPv5ExN6lMrzf82mHXttTqzdSitc0,5863
 zenml/zen_stores/schemas/run_metadata_schemas.py,sha256=pWu-WO0PCITPY5gcIB1uqGBWDT51J7LDzMrOLtNkAGg,6033
-zenml/zen_stores/schemas/run_template_schemas.py,sha256=cT90S8gRYkijrMLqBK6WgpQC1c6LbBnOtMpAgAxhwrg,8768
+zenml/zen_stores/schemas/run_template_schemas.py,sha256=4uE7lHkIQROJRNpNdvATwEJMkeij6pXnHvOsT3q3VrI,8826
 zenml/zen_stores/schemas/schedule_schema.py,sha256=w3tAHt2mEQ7ILPz4ogRdLV4KivB7E2LtIZWkzXLzAIU,7354
 zenml/zen_stores/schemas/schema_utils.py,sha256=2qPNPfasnTmCW1RPeViXoTAp6oO-uRnUpEZ_rugcoNw,2612
 zenml/zen_stores/schemas/secret_schemas.py,sha256=nRzFkgR940BbS1nlBs4H3YuvFCiGGYZ1vQIsu3Y0Mvk,9884
@@ -1396,11 +1396,11 @@ zenml/zen_stores/secrets_stores/hashicorp_secrets_store.py,sha256=NfW1EHIA99lseb
 zenml/zen_stores/secrets_stores/secrets_store_interface.py,sha256=Q2Jbnt2Pp7NGlR-u1YBfRZV2g8su2Fd0ArBMdksAE-Q,2819
 zenml/zen_stores/secrets_stores/service_connector_secrets_store.py,sha256=kPYX-Z_OOhZCI1CP77ncfV7IsV4e8brklnTXmKxZYNc,7078
 zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=Bq1djrUP9saoD7vECjS7-TlA_7sjJGgw1talri4spjU,8656
-zenml/zen_stores/sql_zen_store.py,sha256=4z458K5ItHkAO0bEUAl6rWJDHKh6GS-sRrt2RPridnE,392435
+zenml/zen_stores/sql_zen_store.py,sha256=Sfw5nql38ZGfPbdIUNLMYsVrfMP1faRGMj0f1r1IYnk,392523
 zenml/zen_stores/template_utils.py,sha256=UB_CBoMIfBequa8g3H7aTI6V_ke1SJr4IW1qwFCL1jc,8932
 zenml/zen_stores/zen_store_interface.py,sha256=kzR_i8vHjULld3MquSaMorcab8lJk1e9RZquw1VXjHY,93510
-zenml_nightly-0.66.0.dev20240910.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.66.0.dev20240910.dist-info/METADATA,sha256=HXKwlZMRzcjTm7ssSRK6JClUhT80eOXfXTf9BmBvR-s,20961
-zenml_nightly-0.66.0.dev20240910.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zenml_nightly-0.66.0.dev20240910.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.66.0.dev20240910.dist-info/RECORD,,
+zenml_nightly-0.66.0.dev20240918.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.66.0.dev20240918.dist-info/METADATA,sha256=UP3HTafnsWOeUXj64GD8CVeQeTeZGh6ekWMfQGT9whw,20961
+zenml_nightly-0.66.0.dev20240918.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zenml_nightly-0.66.0.dev20240918.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.66.0.dev20240918.dist-info/RECORD,,