zenml-nightly 0.83.0.dev20250604__py3-none-any.whl → 0.83.0.dev20250606__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.83.0.dev20250604
+0.83.0.dev20250606

zenml/artifacts/utils.py

@@ -37,6 +37,7 @@ from zenml.artifacts.preexisting_data_materializer import (
 )
 from zenml.client import Client
 from zenml.constants import (
+    ENV_ZENML_SERVER,
     MODEL_METADATA_YAML_FILE_NAME,
 )
 from zenml.enums import (
@@ -820,11 +821,18 @@ def _load_artifact_store(
             an artifact store.
         NotImplementedError: If the artifact store could not be loaded.
     """
+    client = Client()
     if isinstance(artifact_store_id, str):
         artifact_store_id = UUID(artifact_store_id)
 
+    if (
+        ENV_ZENML_SERVER not in os.environ
+        and client.active_stack.artifact_store.id == artifact_store_id
+    ):
+        return client.active_stack.artifact_store
+
     if zen_store is None:
-        zen_store = Client().zen_store
+        zen_store = client.zen_store
 
     try:
         artifact_store_model = zen_store.get_stack_component(artifact_store_id)
@@ -857,9 +865,18 @@ def _load_artifact_store(
 def _get_artifact_store_from_response_or_from_active_stack(
     artifact: ArtifactVersionResponse,
 ) -> "BaseArtifactStore":
+    client = Client()
+
     if artifact.artifact_store_id:
+        if (
+            ENV_ZENML_SERVER not in os.environ
+            and client.active_stack.artifact_store.id
+            == artifact.artifact_store_id
+        ):
+            return client.active_stack.artifact_store
+
         try:
-            artifact_store_model = Client().get_stack_component(
+            artifact_store_model = client.get_stack_component(
                 component_type=StackComponentType.ARTIFACT_STORE,
                 name_id_or_prefix=artifact.artifact_store_id,
             )
@@ -881,7 +898,8 @@ def _get_artifact_store_from_response_or_from_active_stack(
                 "the environment that you are loading this artifact from "
                 "has the right dependencies."
             )
-    return Client().active_stack.artifact_store
+
+    return client.active_stack.artifact_store
 
 
 def _load_file_from_artifact_store(

zenml/client.py

@@ -1480,9 +1480,6 @@ class Client(metaclass=ClientMetaClass):
 
         Returns:
             The model of the active stack for this client.
-
-        Raises:
-            RuntimeError: If the active stack is not set.
         """
         if env_stack_id := os.environ.get(ENV_ZENML_ACTIVE_STACK_ID):
             if not self._active_stack or self._active_stack.id != UUID(
@@ -1509,12 +1506,6 @@ class Client(metaclass=ClientMetaClass):
 
             stack_id = GlobalConfiguration().get_active_stack_id()
 
-        if not stack_id:
-            raise RuntimeError(
-                "No active stack is configured. Run "
-                "`zenml stack set STACK_NAME` to set the active stack."
-            )
-
         return self.get_stack(stack_id)
 
     def activate_stack(
@@ -1550,9 +1541,6 @@ class Client(metaclass=ClientMetaClass):
 
         Args:
             stack: The stack to validate.
-
-        Raises:
-            ValidationError: If the stack configuration is invalid.
         """
         local_components: List[str] = []
         remote_components: List[str] = []
@@ -1617,13 +1605,6 @@ class Client(metaclass=ClientMetaClass):
                 f"local resources."
             )
 
-        if not stack.is_valid:
-            raise ValidationError(
-                "Stack configuration is invalid. A valid"
-                "stack must contain an Artifact Store and "
-                "an Orchestrator."
-            )
-
     # ----------------------------- Services -----------------------------------
 
     def create_service(

zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py

@@ -20,24 +20,35 @@ from typing import (
     Any,
     Dict,
     Iterator,
+    List,
     Optional,
     Tuple,
     Type,
+    Union,
     cast,
 )
 from uuid import UUID
 
 import boto3
-import sagemaker
 from botocore.exceptions import WaiterError
+from sagemaker.estimator import Estimator
+from sagemaker.inputs import TrainingInput
 from sagemaker.network import NetworkConfig
-from sagemaker.processing import ProcessingInput, ProcessingOutput
+from sagemaker.processing import ProcessingInput, ProcessingOutput, Processor
+from sagemaker.session import Session
+from sagemaker.workflow.entities import PipelineVariable
 from sagemaker.workflow.execution_variables import (
-    ExecutionVariable,
     ExecutionVariables,
 )
 from sagemaker.workflow.pipeline import Pipeline
-from sagemaker.workflow.steps import ProcessingStep, TrainingStep
+from sagemaker.workflow.step_collections import (
+    StepCollection,
+)
+from sagemaker.workflow.steps import (
+    ProcessingStep,
+    Step,
+    TrainingStep,
+)
 from sagemaker.workflow.triggers import PipelineSchedule
 
 from zenml.client import Client
@@ -215,7 +226,7 @@ class SagemakerOrchestrator(ContainerizedOrchestrator):
         """
         return SagemakerOrchestratorSettings
 
-    def _get_sagemaker_session(self) -> sagemaker.Session:
+    def _get_sagemaker_session(self) -> Session:
         """Method to create the sagemaker session with proper authentication.
 
         Returns:
@@ -258,7 +269,7 @@ class SagemakerOrchestrator(ContainerizedOrchestrator):
                     aws_session_token=credentials["SessionToken"],
                     region_name=self.config.region,
                 )
-        return sagemaker.Session(
+        return Session(
             boto_session=boto_session, default_bucket=self.config.bucket
         )
 
@@ -324,12 +335,8 @@ class SagemakerOrchestrator(ContainerizedOrchestrator):
                 SagemakerOrchestratorSettings, self.get_settings(step)
             )
 
-            environment[ENV_ZENML_SAGEMAKER_RUN_ID] = (
-                ExecutionVariables.PIPELINE_EXECUTION_ARN
-            )
-
             if step_settings.environment:
-                step_environment = step_settings.environment.copy()
+                split_environment = step_settings.environment.copy()
                 # Sagemaker does not allow environment variables longer than 256
                 # characters to be passed to Processor steps. If an environment variable
                 # is longer than 256 characters, we split it into multiple environment
@@ -337,9 +344,9 @@ class SagemakerOrchestrator(ContainerizedOrchestrator):
                 # custom entrypoint configuration.
                 split_environment_variables(
                     size_limit=SAGEMAKER_PROCESSOR_STEP_ENV_VAR_SIZE_LIMIT,
-                    env=step_environment,
+                    env=split_environment,
                 )
-                environment.update(step_environment)
+                environment.update(split_environment)
 
             use_training_step = (
                 step_settings.use_training_step
@@ -421,28 +428,51 @@ class SagemakerOrchestrator(ContainerizedOrchestrator):
                     )
 
             # Construct S3 inputs to container for step
-            inputs = None
+            training_inputs: Optional[
+                Union[TrainingInput, Dict[str, TrainingInput]]
+            ] = None
+            processing_inputs: Optional[List[ProcessingInput]] = None
 
             if step_settings.input_data_s3_uri is None:
                 pass
             elif isinstance(step_settings.input_data_s3_uri, str):
-                inputs = [
-                    ProcessingInput(
-                        source=step_settings.input_data_s3_uri,
-                        destination="/opt/ml/processing/input/data",
-                        s3_input_mode=step_settings.input_data_s3_mode,
+                if use_training_step:
+                    training_inputs = TrainingInput(
+                        s3_data=step_settings.input_data_s3_uri,
+                        input_mode=step_settings.input_data_s3_mode,
                     )
-                ]
-            elif isinstance(step_settings.input_data_s3_uri, dict):
-                inputs = []
-                for channel, s3_uri in step_settings.input_data_s3_uri.items():
-                    inputs.append(
+                else:
+                    processing_inputs = [
                         ProcessingInput(
-                            source=s3_uri,
-                            destination=f"/opt/ml/processing/input/data/{channel}",
+                            source=step_settings.input_data_s3_uri,
+                            destination="/opt/ml/processing/input/data",
                             s3_input_mode=step_settings.input_data_s3_mode,
                         )
-                    )
+                    ]
+            elif isinstance(step_settings.input_data_s3_uri, dict):
+                if use_training_step:
+                    training_inputs = {}
+                    for (
+                        channel,
+                        s3_uri,
+                    ) in step_settings.input_data_s3_uri.items():
+                        training_inputs[channel] = TrainingInput(
+                            s3_data=s3_uri,
+                            input_mode=step_settings.input_data_s3_mode,
+                        )
+                else:
+                    processing_inputs = []
+                    for (
+                        channel,
+                        s3_uri,
+                    ) in step_settings.input_data_s3_uri.items():
+                        processing_inputs.append(
+                            ProcessingInput(
+                                source=s3_uri,
+                                destination=f"/opt/ml/processing/input/data/{channel}",
+                                s3_input_mode=step_settings.input_data_s3_mode,
+                            )
+                        )
 
             # Construct S3 outputs from container for step
             outputs = None
@@ -475,42 +505,56 @@ class SagemakerOrchestrator(ContainerizedOrchestrator):
                         )
                     )
 
-            # Convert environment to a dict of strings
-            environment = {
+            step_environment: Dict[str, Union[str, PipelineVariable]] = {
                 key: str(value)
-                if not isinstance(value, ExecutionVariable)
+                if not isinstance(value, PipelineVariable)
                 else value
                 for key, value in environment.items()
             }
 
+            step_environment[ENV_ZENML_SAGEMAKER_RUN_ID] = (
+                ExecutionVariables.PIPELINE_EXECUTION_ARN
+            )
+
+            sagemaker_step: Step
             if use_training_step:
                 # Create Estimator and TrainingStep
-                estimator = sagemaker.estimator.Estimator(
+                estimator = Estimator(
                     keep_alive_period_in_seconds=step_settings.keep_alive_period_in_seconds,
                     output_path=output_path,
-                    environment=environment,
+                    environment=step_environment,
                     container_entry_point=entrypoint,
                     **args_for_step_executor,
                 )
+
                 sagemaker_step = TrainingStep(
                     name=step_name,
-                    depends_on=step.spec.upstream_steps,
-                    inputs=inputs,
+                    depends_on=cast(
+                        Optional[List[Union[str, Step, StepCollection]]],
+                        step.spec.upstream_steps,
+                    ),
+                    inputs=training_inputs,
                     estimator=estimator,
                 )
             else:
                 # Create Processor and ProcessingStep
-                processor = sagemaker.processing.Processor(
-                    entrypoint=entrypoint,
-                    env=environment,
+                processor = Processor(
+                    entrypoint=cast(
+                        Optional[List[Union[str, PipelineVariable]]],
+                        entrypoint,
+                    ),
+                    env=step_environment,
                     **args_for_step_executor,
                 )
 
                 sagemaker_step = ProcessingStep(
                     name=step_name,
                     processor=processor,
-                    depends_on=step.spec.upstream_steps,
-                    inputs=inputs,
+                    depends_on=cast(
+                        Optional[List[Union[str, Step, StepCollection]]],
+                        step.spec.upstream_steps,
+                    ),
+                    inputs=processing_inputs,
                     outputs=outputs,
                 )
 

zenml/integrations/aws/step_operators/sagemaker_step_operator.py

@@ -13,10 +13,21 @@
 #  permissions and limitations under the License.
 """Implementation of the Sagemaker Step Operator."""
 
-from typing import TYPE_CHECKING, Dict, List, Optional, Tuple, Type, cast
+from typing import (
+    TYPE_CHECKING,
+    Dict,
+    List,
+    Optional,
+    Tuple,
+    Type,
+    Union,
+    cast,
+)
 
 import boto3
-import sagemaker
+from sagemaker.estimator import Estimator
+from sagemaker.inputs import TrainingInput
+from sagemaker.session import Session
 
 from zenml.client import Client
 from zenml.config.build_configuration import BuildConfiguration
@@ -216,7 +227,7 @@ class SagemakerStepOperator(BaseStepOperator):
         else:
             boto_session = boto3.Session()
 
-        session = sagemaker.Session(
+        session = Session(
             boto_session=boto_session, default_bucket=self.config.bucket
         )
 
@@ -232,9 +243,7 @@ class SagemakerStepOperator(BaseStepOperator):
         estimator_args["sagemaker_session"] = session
 
         # Create Estimator
-        estimator = sagemaker.estimator.Estimator(
-            image_name, self.config.role, **estimator_args
-        )
+        estimator = Estimator(image_name, self.config.role, **estimator_args)
 
         # SageMaker allows 63 characters at maximum for job name - ZenML uses 60 for safety margin.
         step_name = Client().get_run_step(info.step_run_id).name
@@ -248,18 +257,14 @@ class SagemakerStepOperator(BaseStepOperator):
         )
 
         # Construct training input object, if necessary
-        inputs = None
+        inputs: Optional[Union[TrainingInput, Dict[str, TrainingInput]]] = None
 
         if isinstance(settings.input_data_s3_uri, str):
-            inputs = sagemaker.inputs.TrainingInput(
-                s3_data=settings.input_data_s3_uri
-            )
+            inputs = TrainingInput(s3_data=settings.input_data_s3_uri)
         elif isinstance(settings.input_data_s3_uri, dict):
             inputs = {}
             for channel, s3_uri in settings.input_data_s3_uri.items():
-                inputs[channel] = sagemaker.inputs.TrainingInput(
-                    s3_data=s3_uri
-                )
+                inputs[channel] = TrainingInput(s3_data=s3_uri)
 
         experiment_config = {}
         if settings.experiment_name:

zenml/models/v2/core/stack.py

@@ -27,7 +27,7 @@ from typing import (
 )
 from uuid import UUID
 
-from pydantic import Field, model_validator
+from pydantic import Field, field_validator, model_validator
 from sqlmodel import and_
 
 from zenml.constants import STR_FIELD_MAX_LENGTH
@@ -94,18 +94,31 @@ class StackRequest(UserScopedRequest):
         "scratch.",
     )
 
-    @property
-    def is_valid(self) -> bool:
-        """Check if the stack is valid.
+    @field_validator("components")
+    def _validate_components(
+        cls, value: Dict[StackComponentType, List[Union[UUID, ComponentInfo]]]
+    ) -> Dict[StackComponentType, List[Union[UUID, ComponentInfo]]]:
+        """Validate the components of the stack.
+
+        Args:
+            value: The components of the stack.
+
+        Raises:
+            ValueError: If the stack does not contain an orchestrator and
+                artifact store.
 
         Returns:
-            True if the stack is valid, False otherwise.
+            The components of the stack.
         """
-        if not self.components:
-            return False
-        return (
-            StackComponentType.ARTIFACT_STORE in self.components
-            and StackComponentType.ORCHESTRATOR in self.components
+        if value:
+            artifact_stores = value.get(StackComponentType.ARTIFACT_STORE, [])
+            orchestrators = value.get(StackComponentType.ORCHESTRATOR, [])
+
+            if orchestrators and artifact_stores:
+                return value
+
+        raise ValueError(
+            "Stack must contain at least an orchestrator and artifact store."
         )
 
     @model_validator(mode="after")
@@ -163,6 +176,39 @@ class StackUpdate(BaseUpdate):
         title="The stack labels.",
     )
 
+    @field_validator("components")
+    def _validate_components(
+        cls,
+        value: Optional[
+            Dict[StackComponentType, List[Union[UUID, ComponentInfo]]]
+        ],
+    ) -> Optional[Dict[StackComponentType, List[Union[UUID, ComponentInfo]]]]:
+        """Validate the components of the stack.
+
+        Args:
+            value: The components of the stack.
+
+        Raises:
+            ValueError: If the stack does not contain an orchestrator and
+                artifact store.
+
+        Returns:
+            The components of the stack.
+        """
+        if value is None:
+            return None
+
+        if value:
+            artifact_stores = value.get(StackComponentType.ARTIFACT_STORE, [])
+            orchestrators = value.get(StackComponentType.ORCHESTRATOR, [])
+
+            if orchestrators and artifact_stores:
+                return value
+
+        raise ValueError(
+            "Stack must contain at least an orchestrator and artifact store."
+        )
+
 
 # ------------------ Response Model ------------------
 

zenml/zen_stores/sql_zen_store.py

@@ -8925,6 +8925,7 @@ class SqlZenStore(BaseZenStore):
 
         pipeline_run = session.exec(
             select(PipelineRunSchema)
+            .with_for_update()
             .options(
                 joinedload(
                     jl_arg(PipelineRunSchema.deployment), innerjoin=True
@@ -8950,96 +8951,100 @@ class SqlZenStore(BaseZenStore):
             num_steps=num_steps,
         )
 
-        if pipeline_run.is_placeholder_run() and not new_status.is_finished:
-            # If the pipeline run is a placeholder run, no step has been started
-            # for the run yet. This means the orchestrator hasn't started
+        if new_status == pipeline_run.status or (
+            pipeline_run.is_placeholder_run() and not new_status.is_finished
+        ):
+            # The status hasn't changed -> no need to update the status.
+            # If the pipeline run is a placeholder run (=no step has been started
+            # for the run yet), this means the orchestrator hasn't started
             # running yet, and this method is most likely being called as
             # part of the creation of some cached steps. In this case, we don't
             # update the status unless the run is finished.
+
+            # Commit so that we release the lock on the pipeline run.
+            session.commit()
             return
 
-        if new_status != pipeline_run.status:
-            run_update = PipelineRunUpdate(status=new_status)
-            if new_status in {
-                ExecutionStatus.COMPLETED,
-                ExecutionStatus.FAILED,
-            }:
-                run_update.end_time = utc_now()
-                if pipeline_run.start_time and isinstance(
-                    pipeline_run.start_time, datetime
-                ):
-                    duration_time = (
-                        run_update.end_time - pipeline_run.start_time
-                    )
-                    duration_seconds = duration_time.total_seconds()
-                    start_time_str = pipeline_run.start_time.strftime(
+        run_update = PipelineRunUpdate(status=new_status)
+        if new_status.is_finished:
+            run_update.end_time = utc_now()
+
+        pipeline_run.update(run_update)
+        session.add(pipeline_run)
+        # Commit so that we release the lock on the pipeline run.
+        session.commit()
+
+        if new_status.is_finished:
+            assert run_update.end_time
+            if pipeline_run.start_time:
+                duration_time = run_update.end_time - pipeline_run.start_time
+                duration_seconds = duration_time.total_seconds()
+                start_time_str = pipeline_run.start_time.strftime(
+                    "%Y-%m-%dT%H:%M:%S.%fZ"
+                )
+            else:
+                start_time_str = None
+                duration_seconds = None
+
+            stack = pipeline_run.deployment.stack
+            assert stack
+            stack_metadata = {
+                str(component.type): component.flavor
+                for component in stack.components
+            }
+            with track_handler(
+                AnalyticsEvent.RUN_PIPELINE_ENDED
+            ) as analytics_handler:
+                analytics_handler.metadata = {
+                    "project_id": pipeline_run.project_id,
+                    "pipeline_run_id": pipeline_run_id,
+                    "template_id": pipeline_run.deployment.template_id,
+                    "status": new_status,
+                    "num_steps": num_steps,
+                    "start_time": start_time_str,
+                    "end_time": run_update.end_time.strftime(
                         "%Y-%m-%dT%H:%M:%S.%fZ"
-                    )
-                else:
-                    start_time_str = None
-                    duration_seconds = None
-
-                stack = pipeline_run.deployment.stack
-                assert stack
-                stack_metadata = {
-                    str(component.type): component.flavor
-                    for component in stack.components
+                    ),
+                    "duration_seconds": duration_seconds,
+                    **stack_metadata,
                 }
-                with track_handler(
-                    AnalyticsEvent.RUN_PIPELINE_ENDED
-                ) as analytics_handler:
-                    analytics_handler.metadata = {
-                        "project_id": pipeline_run.project_id,
-                        "pipeline_run_id": pipeline_run_id,
-                        "template_id": pipeline_run.deployment.template_id,
-                        "status": new_status,
-                        "num_steps": num_steps,
-                        "start_time": start_time_str,
-                        "end_time": run_update.end_time.strftime(
-                            "%Y-%m-%dT%H:%M:%S.%fZ"
-                        ),
-                        "duration_seconds": duration_seconds,
-                        **stack_metadata,
-                    }
 
-                completed_onboarding_steps: Set[str] = {
-                    OnboardingStep.PIPELINE_RUN,
-                    OnboardingStep.STARTER_SETUP_COMPLETED,
-                }
-                if stack_metadata["orchestrator"] not in {
-                    "local",
-                    "local_docker",
-                }:
-                    completed_onboarding_steps.update(
-                        {
-                            OnboardingStep.PIPELINE_RUN_WITH_REMOTE_ORCHESTRATOR,
-                        }
+            completed_onboarding_steps: Set[str] = {
+                OnboardingStep.PIPELINE_RUN,
+                OnboardingStep.STARTER_SETUP_COMPLETED,
+            }
+            if stack_metadata["orchestrator"] not in {
+                "local",
+                "local_docker",
+            }:
+                completed_onboarding_steps.update(
+                    {
+                        OnboardingStep.PIPELINE_RUN_WITH_REMOTE_ORCHESTRATOR,
+                    }
+                )
+            if stack_metadata["artifact_store"] != "local":
+                completed_onboarding_steps.update(
+                    {
+                        OnboardingStep.PIPELINE_RUN_WITH_REMOTE_ARTIFACT_STORE,
+                        OnboardingStep.PRODUCTION_SETUP_COMPLETED,
+                    }
+                )
+            if OnboardingStep.THIRD_PIPELINE_RUN not in (
+                self._cached_onboarding_state or {}
+            ):
+                onboarding_state = self.get_onboarding_state()
+                if OnboardingStep.PIPELINE_RUN in onboarding_state:
+                    completed_onboarding_steps.add(
+                        OnboardingStep.SECOND_PIPELINE_RUN
                     )
-                if stack_metadata["artifact_store"] != "local":
-                    completed_onboarding_steps.update(
-                        {
-                            OnboardingStep.PIPELINE_RUN_WITH_REMOTE_ARTIFACT_STORE,
-                            OnboardingStep.PRODUCTION_SETUP_COMPLETED,
-                        }
+                if OnboardingStep.SECOND_PIPELINE_RUN in onboarding_state:
+                    completed_onboarding_steps.add(
+                        OnboardingStep.THIRD_PIPELINE_RUN
                     )
-                if OnboardingStep.THIRD_PIPELINE_RUN not in (
-                    self._cached_onboarding_state or {}
-                ):
-                    onboarding_state = self.get_onboarding_state()
-                    if OnboardingStep.PIPELINE_RUN in onboarding_state:
-                        completed_onboarding_steps.add(
-                            OnboardingStep.SECOND_PIPELINE_RUN
-                        )
-                    if OnboardingStep.SECOND_PIPELINE_RUN in onboarding_state:
-                        completed_onboarding_steps.add(
-                            OnboardingStep.THIRD_PIPELINE_RUN
-                        )
 
-                self._update_onboarding_state(
-                    completed_steps=completed_onboarding_steps, session=session
-                )
-            pipeline_run.update(run_update)
-            session.add(pipeline_run)
+            self._update_onboarding_state(
+                completed_steps=completed_onboarding_steps, session=session
+            )
 
     # --------------------------- Triggers ---------------------------
 

{zenml_nightly-0.83.0.dev20250604.dist-info → zenml_nightly-0.83.0.dev20250606.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.83.0.dev20250604
+Version: 0.83.0.dev20250606
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops

{zenml_nightly-0.83.0.dev20250604.dist-info → zenml_nightly-0.83.0.dev20250606.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=vOk95Zv12MpjL0GyiQF6b_h7g_FKv7O4bvZHJ5fJgHA,19
+zenml/VERSION,sha256=dOlZWzCTAp-20wgwKM0CFZujsv4yk51NATYHpGjvRRw,19
 zenml/__init__.py,sha256=CKEyepFK-7akXYiMrNVh92Nb01Cjs23w4_YyI6sgdc8,2242
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -25,7 +25,7 @@ zenml/artifacts/external_artifact.py,sha256=7nLANV0vsGC36H1s_B_awX4hnZgXHCGIscQ2
 zenml/artifacts/external_artifact_config.py,sha256=P172p0JOu8Xx1F8RCQut1dnRpt5lpWXGNqMbY-V90sI,3323
 zenml/artifacts/preexisting_data_materializer.py,sha256=dcahDcHUD3Lvn0-6zE2BG84bkyo_ydAgzBWxtbyJJZQ,3325
 zenml/artifacts/unmaterialized_artifact.py,sha256=JNPKq_sNifQx5wP8jEw7TGBEi26zwKirPGlWX9uxbJI,1300
-zenml/artifacts/utils.py,sha256=JEsqAKXsfJoK7TP2ycaVaTBu0O4UwDmQzFTpjoWEyqU,35468
+zenml/artifacts/utils.py,sha256=44wDliyVDmfNQSRpdUwI4JhhsmGP18FF4mWgyEmN0E8,35933
 zenml/cli/__init__.py,sha256=nxq4ifwLV5qT7Qghb42h02XUOcOihBkZp2xBqgiykM8,75670
 zenml/cli/annotator.py,sha256=JRR7_TJOWKyiKGv1kwSjG1Ay6RBWPVgm0X-D0uSBlyE,6976
 zenml/cli/artifact.py,sha256=7lsAS52DroBTFkFWxkyb-lIDOGP5jPL_Se_RDG_2jgg,9564
@@ -55,7 +55,7 @@ zenml/cli/text_utils.py,sha256=bY1GIjoULt1cW2FyrPlMoAXNS2R7cSOjDFEZQqrpVQ8,3553
 zenml/cli/user_management.py,sha256=sNnhaUxH-cHecbZBR1L0mEU0TnLNZHzI6ZBCUSQa7OY,13078
 zenml/cli/utils.py,sha256=vMAb9f6GDfNVGmZWOz9UOyPRpKI3KfnYpRl_w9YUBNE,86501
 zenml/cli/version.py,sha256=nm1iSU_1V6-MUwpMKeXcwFhLYGUMLswvQL67cEuCpxA,3635
-zenml/client.py,sha256=TZB3jIvHILiDGiMZFIJPH_v7LZBQizFh-3tdgng1d6s,293928
+zenml/client.py,sha256=cf4sYrVCx9xbX3ZcQKF2j-jRroCxy-MKUNkEOhZD1yk,293336
 zenml/client_lazy_loader.py,sha256=oyxKvBWVB7k2pHMavdhNEOfR2Vk4IS3XUu43SBzDPsI,7152
 zenml/code_repositories/__init__.py,sha256=W5bDfzAG8OXIKZSV1L-VHuzMcSCYa9qzTdPb3jqfyYw,920
 zenml/code_repositories/base_code_repository.py,sha256=Id6VjbUu8N3ZpNvBGhOgbahtoMiCAtYXed3G7YQ_iAc,5225
@@ -146,12 +146,12 @@ zenml/integrations/aws/flavors/sagemaker_step_operator_flavor.py,sha256=e3locb2O
 zenml/integrations/aws/image_builders/__init__.py,sha256=91hgH1OphG2i-vk-G8N4yKBFIzK89Wu7BK4-T5yOA7E,786
 zenml/integrations/aws/image_builders/aws_image_builder.py,sha256=UcPYYYjJjfsicY3hV4OZeJt552AVmwPZPlv-AsG1g1I,11489
 zenml/integrations/aws/orchestrators/__init__.py,sha256=Wh0Fhtt_uo6YrkvXY9kL0M478FL7XpapjoFreUZbgUg,794
-zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py,sha256=l5mG9YnND8E_znpI7ErkJFjqUnYvqOxfDzGg4El4imE,37564
+zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py,sha256=EfDFT2WG0t-6x1CU0n57BS7YCC0wPA8CfTGpb4-G0Tc,39282
 zenml/integrations/aws/orchestrators/sagemaker_orchestrator_entrypoint_config.py,sha256=WXCWdVojIZxx5_3-g1T95S2vsJ-RLNGcp-V409wgme0,1555
 zenml/integrations/aws/service_connectors/__init__.py,sha256=w2Md40yG89PwmU9eBceh6dGy3XYZ3MKusNAZ51sGOgE,783
 zenml/integrations/aws/service_connectors/aws_service_connector.py,sha256=7H69IoOYmyn5QcXEfL1-OmC0UaQ54TfNNhv2t8A6Daw,92107
 zenml/integrations/aws/step_operators/__init__.py,sha256=HK5oIqLixYKUKoiN4LnqTyGjAZJUdUqWYGqJhFSWyo0,817
-zenml/integrations/aws/step_operators/sagemaker_step_operator.py,sha256=8g9p9eHvZkg2LY2lnWPVkriB7Jr_LbQcw2HQFjVgZ6E,10330
+zenml/integrations/aws/step_operators/sagemaker_step_operator.py,sha256=1RCHSzHcgTldChlk4c8iCrKTeeHrdQlQUJpMepk9ShE,10384
 zenml/integrations/aws/step_operators/sagemaker_step_operator_entrypoint_config.py,sha256=2cXroe6-bXyHVkO5yPnZagNpqx6MrMDcvuvNr8J2j-A,1581
 zenml/integrations/azure/__init__.py,sha256=iJ8smVn76fIuOeeJV5txrBfBS0PyRKUNt9aVkL3Ucic,3103
 zenml/integrations/azure/artifact_stores/__init__.py,sha256=dlIwbpgjE0Hy4rhMbelNJHVKm4t8tj_hRu9mQ_cEIAg,820
@@ -658,7 +658,7 @@ zenml/models/v2/core/server_settings.py,sha256=NNNsYM2AwfsPmA2kfTMwqxn0__9WDoii5
 zenml/models/v2/core/service.py,sha256=PeI036PIVG0zX5EiYPsx7h5LTRC8hlmfdeClKff-IaU,16106
 zenml/models/v2/core/service_account.py,sha256=-1c9Et9Ma4_OHOZxIHjVnLJAaLLxhpoLEyGKkwc1Yx8,6619
 zenml/models/v2/core/service_connector.py,sha256=5JBRqd3iL0SFnupUO8fSgbsAZY9u2FDkMaY44azV5FQ,37048
-zenml/models/v2/core/stack.py,sha256=z5PWFLsyfGi_eOpiOo6Qca98rzAf8_DIfagaXPcy3P0,11829
+zenml/models/v2/core/stack.py,sha256=3IKofEaKprAtYNbIrKvcYs5IABTbCqxcfJSXC3Q0fYM,13332
 zenml/models/v2/core/step_run.py,sha256=NLA9cGk76_nGfxhRuXeEJFv4LzWiIl70AtSsVkJ6NCU,21020
 zenml/models/v2/core/tag.py,sha256=60hCZuHAHy7lsYFkTDw7hK0zZz2eG7jx8l9VY7Hh-LQ,7100
 zenml/models/v2/core/tag_resource.py,sha256=H7CPj9oaUpym8vSkwuYSJ6_rWfHMEKVRedPw2kyxmII,2640
@@ -1324,11 +1324,11 @@ zenml/zen_stores/secrets_stores/hashicorp_secrets_store.py,sha256=5err1a-TrV3SR5
 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=S87ne23D08PAwtfRVlVnBn8R0ilTpEh6r8blauNV5WQ,6941
 zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjsrl2Tf1yo281xUTjSBsos4qk,8788
-zenml/zen_stores/sql_zen_store.py,sha256=L5EhDaO-799UXJZQChGmOos343HJx4S3L16_x_N2gqw,466193
+zenml/zen_stores/sql_zen_store.py,sha256=zgdaIUTG2JYTr6_sW_yhYar_ZgPQZiKWEO3rmM0ys1Y,466071
 zenml/zen_stores/template_utils.py,sha256=GbJ7LgGVYHSCKPEA8RNTxPoVTWqpC77F_lGzjJ4O1Fw,9220
 zenml/zen_stores/zen_store_interface.py,sha256=_ap55L3_mrHgegsLkMRSmmNXVasYC53LwjcEeuS1YT4,92411
-zenml_nightly-0.83.0.dev20250604.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.83.0.dev20250604.dist-info/METADATA,sha256=UC_wYNNxKBdlqijXKERAmru_bMBTX4MUnJq-YIhkUoM,24317
-zenml_nightly-0.83.0.dev20250604.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.83.0.dev20250604.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.83.0.dev20250604.dist-info/RECORD,,
+zenml_nightly-0.83.0.dev20250606.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.83.0.dev20250606.dist-info/METADATA,sha256=GcnWjjJZNPcjOoYHllarFKXh6zn0RmaoyeAeGwyS9xY,24317
+zenml_nightly-0.83.0.dev20250606.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.83.0.dev20250606.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.83.0.dev20250606.dist-info/RECORD,,