zenml-nightly 0.83.0.dev20250611__py3-none-any.whl → 0.83.0.dev20250613__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.83.0.dev20250611
+0.83.0.dev20250613

zenml/artifacts/utils.py

@@ -852,7 +852,7 @@ def _load_artifact_store(
             StackComponent.from_model(artifact_store_model),
         )
     except ImportError:
-        link = "https://docs.zenml.io/stack-components/artifact-stores/custom#enabling-artifact-visualizations-with-custom-artifact-stores"
+        link = "https://docs.zenml.io/stacks/stack-components/artifact-stores/custom#enabling-artifact-visualizations-with-custom-artifact-stores"
         raise NotImplementedError(
             f"Artifact store '{artifact_store_model.name}' could not be "
             f"instantiated. This is likely because the artifact store's "
@@ -950,7 +950,7 @@ def _load_file_from_artifact_store(
         raise e
     except Exception as e:
         logger.exception(e)
-        link = "https://docs.zenml.io/stack-components/artifact-stores/custom#enabling-artifact-visualizations-with-custom-artifact-stores"
+        link = "https://docs.zenml.io/stacks/stack-components/artifact-stores/custom#enabling-artifact-visualizations-with-custom-artifact-stores"
         raise NotImplementedError(
             f"File '{uri}' could not be loaded because the underlying artifact "
             f"store '{artifact_store.name}' could not open the file. This is "

zenml/cli/__init__.py

@@ -80,7 +80,7 @@ You can also pass in a directory path manually using the
 zenml init --path /path/to/dir
 ```
 
-If you wish to use one of [the available ZenML project templates](https://docs.zenml.io/how-to/project-setup-and-management/collaborate-with-team/project-templates)
+If you wish to use one of [the available ZenML project templates](https://docs.zenml.io/user-guides/best-practices/project-templates)
 to generate a ready-to-use project scaffold in your repository, you can do so by
 passing the ``--template`` option:
 
@@ -232,7 +232,7 @@ and `and` keywords.
 Artifact Stores
 ---------------
 
-In ZenML, [the artifact store](https://docs.zenml.io/stack-components/artifact-stores)
+In ZenML, [the artifact store](https://docs.zenml.io/stacks/stack-components/artifact-stores)
 is where all the inputs and outputs of your pipeline steps are stored. By
 default, ZenML initializes your repository with an artifact store with
 everything kept on your local machine. You can get a better understanding
@@ -327,7 +327,7 @@ zenml artifact-store --help
 Orchestrators
 -------------
 
-An [orchestrator](https://docs.zenml.io/stack-components/orchestrators)
+An [orchestrator](https://docs.zenml.io/stacks/stack-components/orchestrators)
 is a special kind of backend that manages the running of each step of the
 pipeline. Orchestrators administer the actual pipeline runs. By default,
 ZenML initializes your repository with an orchestrator that runs everything
@@ -422,7 +422,7 @@ zenml orchestrators --help
 Container Registries
 --------------------
 
-[The container registry](https://docs.zenml.io/stack-components/container-registries)
+[The container registry](https://docs.zenml.io/stacks/stack-components/container-registries)
 is where all the images that are used by a container-based orchestrator are
 stored. To get a better understanding regarding container registries, use
 the command:
@@ -517,7 +517,7 @@ zenml container-registry --help
 Data Validators
 ---------------
 
-In ZenML, [data validators](https://docs.zenml.io/stack-components/data-validators)
+In ZenML, [data validators](https://docs.zenml.io/stacks/stack-components/data-validators)
 help you profile and validate your data.
 
 By default, a default ZenML local stack will not register a data validator. If
@@ -604,7 +604,7 @@ zenml data-validator --help
 Experiment Trackers
 -------------------
 
-[Experiment trackers](https://docs.zenml.io/stack-components/experiment-trackers)
+[Experiment trackers](https://docs.zenml.io/stacks/stack-components/experiment-trackers)
  let you track your ML experiments by logging the parameters
 and allow you to compare between different runs. To get a better
 understanding regarding experiment trackers, use the command:
@@ -700,7 +700,7 @@ zenml experiment-tracker --help
 Model Deployers
 ---------------
 
-[Model deployers](https://docs.zenml.io/stack-components/model-deployers)
+[Model deployers](https://docs.zenml.io/stacks/stack-components/model-deployers)
 are stack components responsible for online model serving. They are responsible
 for deploying models to a remote server. Model deployers also act as a registry
 for models that are served with ZenML. To get a better understanding regarding
@@ -837,7 +837,7 @@ zenml model-deployer --help
 Step Operators
 --------------
 
-[Step operators](https://docs.zenml.io/stack-components/step-operators)
+[Step operators](https://docs.zenml.io/stacks/stack-components/step-operators)
 allow you to run individual steps in a custom environment different from the
 default one used by your active orchestrator. One example use-case is to run a
 training step of your pipeline in an environment with GPUs available. To get
@@ -932,7 +932,7 @@ zenml step-operator --help
 Alerters
 --------
 
-In ZenML, [alerters](https://docs.zenml.io/stack-components/alerters)
+In ZenML, [alerters](https://docs.zenml.io/stacks/stack-components/alerters)
 allow you to send alerts from within your pipeline.
 
 By default, a default ZenML local stack will not register an alerter. If
@@ -1017,7 +1017,7 @@ zenml alerter --help
 Feature Stores
 --------------
 
-[Feature stores](https://docs.zenml.io/stack-components/feature-stores)
+[Feature stores](https://docs.zenml.io/stacks/stack-components/feature-stores)
 allow data teams to serve data via an offline store and an online low-latency
 store where data is kept in sync between the two. To get a better understanding
 regarding feature stores, use the command:
@@ -1114,7 +1114,7 @@ zenml feature-store --help
 Annotators
 ----------
 
-[Annotators](https://docs.zenml.io/stack-components/annotators)
+[Annotators](https://docs.zenml.io/stacks/stack-components/annotators)
 enable the use of data annotation as part of your ZenML stack and pipelines.
 
 By default, a default ZenML local stack will not register an annotator. If
@@ -1206,7 +1206,7 @@ zenml annotator --help
 Image Builders
 --------------
 
-In ZenML, [image builders](https://docs.zenml.io/stack-components/image-builders)
+In ZenML, [image builders](https://docs.zenml.io/stacks/stack-components/image-builders)
 allow you to build container images such
 that your machine-learning pipelines and steps can be executed in remote
 environments.
@@ -1295,7 +1295,7 @@ zenml image-builder --help
 Model Registries
 ----------------
 
-[Model registries](https://docs.zenml.io/stack-components/model-registries)
+[Model registries](https://docs.zenml.io/stacks/stack-components/model-registries)
 are centralized repositories that facilitate the collaboration and management
 of machine learning models. To get a better understanding regarding model
 registries as a concept, use the command:
@@ -1985,7 +1985,7 @@ Secrets management
 ------------------
 
 ZenML offers a way to [securely store secrets associated with your other
-stack components and infrastructure](https://docs.zenml.io/getting-started/deploying-zenml/secret-management).
+stack components and infrastructure](https://docs.zenml.io/deploying-zenml/deploying-zenml/secret-management).
 A ZenML Secret is a collection or grouping of key-value pairs stored by the
 ZenML secrets store. ZenML Secrets are identified by a unique name which
 allows you to fetch or reference them in your pipelines and stacks.
@@ -2082,7 +2082,7 @@ challenge in configuring uninterrupted, secure access to infrastructure
 resources. In ZenML, Service Connectors streamline this process by abstracting
 away the complexity of authentication and help you connect your stack to your
 resources. You can find the full docs on the ZenML service connectors
-[here](https://docs.zenml.io/how-to/infrastructure-deployment/auth-management).
+[here](https://docs.zenml.io/stacks/service-connectors/auth-management).
 
 The ZenML CLI features a variety of commands to help you manage your service
 connectors. First of all, to explore all the types of service connectors
@@ -2112,7 +2112,7 @@ zenml service-connector register SERVICE_CONNECTOR_NAME \
 ```
 
 For more details on how to create a service connector, please refer to our
-[docs](https://docs.zenml.io/how-to/infrastructure-deployment/auth-management).
+[docs](https://docs.zenml.io/stacks/service-connectors/auth-management).
 
 To check if your service connector is registered properly, you can `verify` it.
 By doing this, you can both check if it is configured correctly and also, you
@@ -2366,7 +2366,7 @@ defining the pipeline is not in your current directory, the module path consists
 of the full path to the file, separated by dots, e.g.
 `some_directory.some_file.my_pipeline`.
 
-To [build Docker images for your pipeline](https://docs.zenml.io/how-to/infrastructure-deployment/customize-docker-builds)
+To [build Docker images for your pipeline](https://docs.zenml.io/concepts/containerization)
 without actually running the pipeline, use:
 
 ```bash
@@ -2429,7 +2429,7 @@ Tagging your resources with ZenML
 ---------------------------------
 
 When you are using ZenML, you can [use tags to organize and categorize your
-assets](https://docs.zenml.io/how-to/handle-data-artifacts/tagging).
+assets](https://docs.zenml.io/concepts/tags).
 This way, you can streamline your workflows and enhance the discoverability of
 your resources more easily.
 

zenml/cli/base.py

@@ -264,7 +264,7 @@ def init(
         f"will only take effect when you're running ZenML from the initialized "
         f"repository root, or from a subdirectory. For more information on "
         f"repositories and configurations, please visit "
-        f"https://docs.zenml.io/user-guides/production-guide/understand-stacks."
+        f"https://docs.zenml.io/user-guides/production-guide/understand-stacks"
     )
 
 

zenml/cli/server.py

@@ -413,7 +413,7 @@ def connect(
             "filesystem and is no longer supported. The web login workflow will "
             "be used instead. An alternative for non-interactive environments "
             "is to create and use a service account API key (see "
-            "https://docs.zenml.io/how-to/manage-zenml-server/connecting-to-zenml/connect-with-a-service-account "
+            "https://docs.zenml.io/deploying-zenml/connecting-to-zenml/connect-with-a-service-account "
             "for more information)."
         )
 

zenml/cli/stack.py

@@ -276,7 +276,7 @@ def register_stack(
                 "ZenML needs to be accessible from the cloud provider to allow "
                 "the stack and its components to be registered automatically. "
                 "Please deploy ZenML in a remote environment as described in "
-                "the documentation: https://docs.zenml.io/getting-started/deploying-zenml "
+                "the documentation: https://docs.zenml.io/deploying-zenml/deploying-zenml "
                 "or use a managed ZenML Pro server instance for quick access "
                 "to this feature and more: https://www.zenml.io/pro"
             )
@@ -1515,7 +1515,7 @@ def deploy(
             "ZenML needs to be accessible from the cloud provider to allow the "
             "stack and its components to be registered automatically. "
             "Please deploy ZenML in a remote environment as described in the "
-            "documentation: https://docs.zenml.io/getting-started/deploying-zenml "
+            "documentation: https://docs.zenml.io/deploying-zenml/deploying-zenml "
             "or use a managed ZenML Pro server instance for quick access to "
             "this feature and more: https://www.zenml.io/pro"
         )

zenml/client.py

@@ -3542,7 +3542,9 @@ class Client(metaclass=ClientMetaClass):
         """
         return self._get_entity_by_id_or_name_or_prefix(
             get_method=self.zen_store.get_run_template,
-            list_method=self.list_run_templates,
+            list_method=functools.partial(
+                self.list_run_templates, hidden=None
+            ),
             name_id_or_prefix=name_id_or_prefix,
             allow_name_prefix_match=False,
             project=project,

zenml/config/compiler.py

@@ -598,7 +598,7 @@ class Compiler:
                 f"no steps. Please make sure that your steps are decorated "
                 "with `@step` and that at least one step is called within the "
                 "pipeline. For more information, see "
-                "https://docs.zenml.io/user-guides/starter-guide."
+                "https://docs.zenml.io/user-guides/starter-guide"
             )
 
         additional_spec_args: Dict[str, Any] = {

zenml/config/secret_reference_mixin.py

@@ -66,7 +66,7 @@ class SecretReferenceMixin(BaseModel):
                         "but future versions of ZenML will require you to pass "
                         "in sensitive information as secrets. Check out the "
                         "documentation on how to configure values with secrets "
-                        "here: https://docs.zenml.io/getting-started/deploying-zenml/secret-management"
+                        "here: https://docs.zenml.io/deploying-zenml/deploying-zenml/secret-management"
                     )
                 continue
 

zenml/integrations/aws/flavors/sagemaker_orchestrator_flavor.py

@@ -190,7 +190,7 @@ class SagemakerOrchestratorConfig(
         scheduler_role: The ARN of the IAM role that will be assumed by
             the EventBridge service to launch Sagemaker pipelines
             (For more details regarding the required permissions, please check:
-            https://docs.zenml.io/stack-components/orchestrators/sagemaker#required-iam-permissions-for-schedules)
+            https://docs.zenml.io/stacks/stack-components/orchestrators/sagemaker#required-iam-permissions-for-schedules)
         aws_access_key_id: The AWS access key ID to use to authenticate to AWS.
             If not provided, the value from the default AWS config will be used.
         aws_secret_access_key: The AWS secret access key to use to authenticate

zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py

@@ -693,7 +693,7 @@ class SagemakerOrchestrator(ContainerizedOrchestrator):
                         "your client side credentials that you are "
                         "is not configured correctly to schedule sagemaker "
                         "pipelines. For more information, please check:"
-                        "https://docs.zenml.io/stack-components/orchestrators/sagemaker#required-iam-permissions-for-schedules"
+                        "https://docs.zenml.io/stacks/stack-components/orchestrators/sagemaker#required-iam-permissions-for-schedules"
                     )
             else:
                 scheduler_role_arn = self.config.scheduler_role

zenml/integrations/gcp/orchestrators/vertex_orchestrator.py

@@ -913,7 +913,7 @@ class VertexOrchestrator(ContainerizedOrchestrator, GoogleCredentialsMixin):
                     "set or set to 0. The accelerator type will be ignored. "
                     "To fix this warning, either remove the specified "
                     "accelerator type or set the `gpu_count` using the "
-                    "ResourceSettings (https://docs.zenml.io/how-to/advanced-topics/training-with-gpus)."
+                    "ResourceSettings (https://docs.zenml.io/user-guides/tutorial/distributed-training)."
                 )
 
         return dynamic_component

zenml/integrations/neptune/experiment_trackers/run_state.py

@@ -171,7 +171,7 @@ def get_neptune_run() -> "Run":
             "Unable to get neptune run: The experiment tracker has not been "
             "initialized. To solve this, make sure you use the experiment "
             "tracker in your step. See "
-            "https://docs.zenml.io/stack-components/experiment-trackers/neptune#how-do-you-use-it "
+            "https://docs.zenml.io/stacks/stack-components/experiment-trackers/neptune#how-do-you-use-it "
             "for more information."
         )
 

zenml/materializers/base_materializer.py

@@ -68,7 +68,7 @@ class BaseMaterializerMeta(type):
                     f"Invalid materializer class '{name}'. When creating a "
                     f"custom materializer, make sure to specify at least one "
                     f"type in its ASSOCIATED_TYPES class variable.",
-                    url="https://docs.zenml.io/how-to/handle-data-artifacts/handle-custom-data-types",
+                    url="https://docs.zenml.io/concepts/artifacts/materializers",
                 )
 
             # Validate associated artifact type.
@@ -83,7 +83,7 @@ class BaseMaterializerMeta(type):
                         f"custom materializer, make sure to specify a valid "
                         f"artifact type in its ASSOCIATED_ARTIFACT_TYPE class "
                         f"variable.",
-                        url="https://docs.zenml.io/how-to/handle-data-artifacts/handle-custom-data-types",
+                        url="https://docs.zenml.io/concepts/artifacts/materializers",
                     )
 
             # Validate associated data types.
@@ -92,7 +92,7 @@ class BaseMaterializerMeta(type):
                     raise MaterializerInterfaceError(
                         f"Associated type {associated_type} for materializer "
                         f"{name} is not a class.",
-                        url="https://docs.zenml.io/how-to/handle-data-artifacts/handle-custom-data-types",
+                        url="https://docs.zenml.io/concepts/artifacts/materializers",
                     )
 
             # Register the materializer.

zenml/materializers/cloudpickle_materializer.py

@@ -102,7 +102,7 @@ class CloudpickleMaterializer(BaseMaterializer):
                 "the artifacts cannot be loaded when running with a different "
                 "Python version. Please consider implementing a custom "
                 f"materializer for type `{type(data)}` according to the "
-                "instructions at https://docs.zenml.io/how-to/handle-data-artifacts/handle-custom-data-types"
+                "instructions at https://docs.zenml.io/concepts/artifacts/materializers"
             )
 
         # save python version for validation on loading

zenml/model/utils.py

@@ -58,7 +58,7 @@ def log_model_metadata(
         "The `log_model_metadata` function is deprecated and will soon be "
         "removed. Instead, you can consider using: "
         "`log_metadata(metadata={...}, infer_model=True)` instead. For more "
-        "info: https://docs.zenml.io/how-to/model-management-metrics/track-metrics-metadata/attach-metadata-to-a-model"
+        "info: https://docs.zenml.io/concepts/metadata#attaching-metadata-to-models"
     )
 
     from zenml import log_metadata

zenml/orchestrators/base_orchestrator.py

@@ -61,7 +61,7 @@ class BaseOrchestratorConfig(StackComponentConfig):
                     "The 'custom_docker_base_image_name' field has been "
                     "deprecated. To use a custom base container image with your "
                     "orchestrators, please use the DockerSettings in your "
-                    "pipeline (see https://docs.zenml.io/how-to/infrastructure-deployment/customize-docker-builds)."
+                    "pipeline (see https://docs.zenml.io/concepts/containerization)."
                 )
 
         return data

zenml/orchestrators/utils.py

@@ -182,7 +182,7 @@ def get_config_environment_vars(
                         "service account API key to authenticate to the ZenML "
                         "server instead of your regular user account. For more "
                         "information, see "
-                        "https://docs.zenml.io/how-to/manage-zenml-server/connecting-to-zenml/connect-with-a-service-account"
+                        "https://docs.zenml.io/deploying-zenml/connecting-to-zenml/connect-with-a-service-account"
                     )
 
                 # The schedule, pipeline run or step run credentials are scoped to

zenml/pipelines/pipeline_definition.py

@@ -686,7 +686,7 @@ To avoid this consider setting pipeline parameters only in one place (config or
                     f"Stack {stack.name} does not support scheduling. "
                     "Not all orchestrator types support scheduling, "
                     "kindly consult with "
-                    "https://docs.zenml.io/how-to/build-pipelines/schedule-a-pipeline "
+                    "https://docs.zenml.io/concepts/steps_and_pipelines/scheduling "
                     "for details."
                 )
             if schedule.name:

zenml/service_connectors/service_connector_utils.py

@@ -60,9 +60,9 @@ def _raise_specific_cloud_exception_if_needed(
     orchestrators: List[ResourcesInfo],
     container_registries: List[ResourcesInfo],
 ) -> None:
-    AWS_DOCS = "https://docs.zenml.io/how-to/infrastructure-deployment/auth-management/aws-service-connector"
-    GCP_DOCS = "https://docs.zenml.io/how-to/infrastructure-deployment/auth-management/gcp-service-connector"
-    AZURE_DOCS = "https://docs.zenml.io/how-to/infrastructure-deployment/auth-management/azure-service-connector"
+    AWS_DOCS = "https://docs.zenml.io/stacks/service-connectors/connector-types/aws-service-connector"
+    GCP_DOCS = "https://docs.zenml.io/stacks/service-connectors/connector-types/gcp-service-connector"
+    AZURE_DOCS = "https://docs.zenml.io/stacks/service-connectors/connector-types/azure-service-connector"
 
     if not artifact_stores:
         error_msg = (

zenml/stack/stack_component.py

@@ -106,7 +106,7 @@ class StackComponentConfig(BaseModel, ABC):
                         "in sensitive information as secrets. Check out the "
                         "documentation on how to configure your stack "
                         "components with secrets here: "
-                        "https://docs.zenml.io/getting-started/deploying-zenml/secret-management"
+                        "https://docs.zenml.io/deploying-zenml/deploying-zenml/secret-management"
                     )
                 continue
 

zenml/steps/entrypoint_function_utils.py

@@ -83,7 +83,7 @@ class StepArtifact:
             "you're trying to unpack the return value of your step but the "
             "step only returns a single artifact. For more information on how "
             "to add type annotations to your step to indicate multiple "
-            "artifacts visit https://docs.zenml.io/how-to/build-pipelines/step-output-typing-and-annotation#type-annotations."
+            "artifacts visit https://docs.zenml.io/concepts/steps_and_pipelines#multiple-return-values."
         )
 
 

zenml/utils/tag_utils.py

@@ -44,7 +44,7 @@ add_tags(tags=[...], run_template=...)
 # Manual tagging to an artifact
 add_tags(tags=[...], artifact=...)
 
-# Automatic tagging to an artifact version(within a step)
+# Automatic tagging to an artifact version (within a step)
 add_tags(tags=[...], infer_artifact=True)  # step with single output
 add_tags(tags=[...], artifact_name=..., infer_artifact=True)  # specific output of a step
 
@@ -148,7 +148,7 @@ def add_tags(
     *,
     tags: List[Union[str, Tag]],
     artifact_name: str,
-    artifact_version: Optional[str] = None,
+    artifact_version: str,
 ) -> None: ...
 
 
@@ -191,7 +191,7 @@ def add_tags(
     artifact_version_id: Optional[UUID] = None,
     artifact_name: Optional[str] = None,
     artifact_version: Optional[str] = None,
-    infer_artifact: bool = False,
+    infer_artifact: Optional[bool] = None,
 ) -> None:
     """Add tags to various resource types in a generalized way.
 
@@ -220,19 +220,52 @@ def add_tags(
     resource_type = None
 
     # Tag a pipeline
-    if pipeline is not None:
+    if pipeline is not None and all(
+        v is None
+        for v in [
+            run,
+            run_template,
+            artifact,
+            artifact_version_id,
+            artifact_name,
+            artifact_version,
+            infer_artifact,
+        ]
+    ):
         pipeline_model = client.get_pipeline(name_id_or_prefix=pipeline)
         resource_id = pipeline_model.id
         resource_type = TaggableResourceTypes.PIPELINE
 
     # Tag a run by ID
-    elif run is not None:
+    elif run is not None and all(
+        v is None
+        for v in [
+            pipeline,
+            run_template,
+            artifact,
+            artifact_version_id,
+            artifact_name,
+            artifact_version,
+            infer_artifact,
+        ]
+    ):
         run_model = client.get_pipeline_run(name_id_or_prefix=run)
         resource_id = run_model.id
         resource_type = TaggableResourceTypes.PIPELINE_RUN
 
     # Tag a run template
-    elif run_template is not None:
+    elif run_template is not None and all(
+        v is None
+        for v in [
+            pipeline,
+            run,
+            artifact,
+            artifact_version_id,
+            artifact_name,
+            artifact_version,
+            infer_artifact,
+        ]
+    ):
         run_template_model = client.get_run_template(
             name_id_or_prefix=run_template
         )
@@ -240,26 +273,68 @@ def add_tags(
         resource_type = TaggableResourceTypes.RUN_TEMPLATE
 
     # Tag an artifact
-    elif artifact is not None:
+    elif artifact is not None and all(
+        v is None
+        for v in [
+            pipeline,
+            run,
+            run_template,
+            artifact_version_id,
+            artifact_name,
+            artifact_version,
+            infer_artifact,
+        ]
+    ):
         artifact_model = client.get_artifact(name_id_or_prefix=artifact)
         resource_id = artifact_model.id
         resource_type = TaggableResourceTypes.ARTIFACT
 
+    # Tag an artifact version by its ID
+    elif artifact_version_id is not None and all(
+        v is None
+        for v in [
+            pipeline,
+            run,
+            run_template,
+            artifact,
+            artifact_name,
+            artifact_version,
+            infer_artifact,
+        ]
+    ):
+        resource_id = artifact_version_id
+        resource_type = TaggableResourceTypes.ARTIFACT_VERSION
+
     # Tag an artifact version by its name and version
-    elif artifact_name is not None and artifact_version is not None:
+    elif (artifact_name is not None and artifact_version is not None) and all(
+        v is None
+        for v in [
+            pipeline,
+            run,
+            run_template,
+            artifact,
+            artifact_version_id,
+            infer_artifact,
+        ]
+    ):
         artifact_version_model = client.get_artifact_version(
             name_id_or_prefix=artifact_name, version=artifact_version
         )
         resource_id = artifact_version_model.id
         resource_type = TaggableResourceTypes.ARTIFACT_VERSION
 
-    # Tag an artifact version by its ID
-    elif artifact_version_id is not None:
-        resource_id = artifact_version_id
-        resource_type = TaggableResourceTypes.ARTIFACT_VERSION
-
     # Tag an artifact version through the step context
-    elif infer_artifact is True:
+    elif infer_artifact is True and all(
+        v is None
+        for v in [
+            pipeline,
+            run,
+            run_template,
+            artifact,
+            artifact_version_id,
+            artifact_version,
+        ]
+    ):
         resource_type = TaggableResourceTypes.ARTIFACT_VERSION
 
         try:
@@ -308,12 +383,14 @@ def add_tags(
     elif all(
         v is None
         for v in [
+            pipeline,
             run,
+            run_template,
+            artifact,
             artifact_version_id,
             artifact_name,
             artifact_version,
-            pipeline,
-            run_template,
+            infer_artifact,
         ]
     ):
         try:
@@ -437,7 +514,7 @@ def remove_tags(
     *,
     tags: List[str],
     artifact_name: str,
-    artifact_version: Optional[str] = None,
+    artifact_version: str,
 ) -> None: ...
 
 
@@ -464,7 +541,7 @@ def remove_tags(
     artifact_version_id: Optional[UUID] = None,
     artifact_name: Optional[str] = None,
     artifact_version: Optional[str] = None,
-    infer_artifact: bool = False,
+    infer_artifact: Optional[bool] = None,
 ) -> None:
     """Remove tags from various resource types in a generalized way.
 
@@ -492,13 +569,35 @@ def remove_tags(
     resource_type = None
 
     # Remove tags from a pipeline
-    if pipeline is not None:
+    if pipeline is not None and all(
+        v is None
+        for v in [
+            run_template,
+            run,
+            artifact,
+            artifact_version_id,
+            artifact_name,
+            artifact_version,
+            infer_artifact,
+        ]
+    ):
         pipeline_model = client.get_pipeline(name_id_or_prefix=pipeline)
         resource_id = pipeline_model.id
         resource_type = TaggableResourceTypes.PIPELINE
 
     # Remove tags from a run template
-    elif run_template is not None:
+    elif run_template is not None and all(
+        v is None
+        for v in [
+            pipeline,
+            run,
+            artifact,
+            artifact_version_id,
+            artifact_version,
+            artifact_name,
+            infer_artifact,
+        ]
+    ):
         run_template_model = client.get_run_template(
             name_id_or_prefix=run_template
         )
@@ -506,32 +605,85 @@ def remove_tags(
         resource_type = TaggableResourceTypes.RUN_TEMPLATE
 
     # Remove tags from a run
-    elif run is not None:
+    elif run is not None and all(
+        v is None
+        for v in [
+            pipeline,
+            run_template,
+            artifact,
+            artifact_version_id,
+            artifact_name,
+            artifact_version,
+            infer_artifact,
+        ]
+    ):
         run_model = client.get_pipeline_run(name_id_or_prefix=run)
         resource_id = run_model.id
         resource_type = TaggableResourceTypes.PIPELINE_RUN
 
     # Remove tags from an artifact
-    elif artifact is not None:
+    elif artifact is not None and all(
+        v is None
+        for v in [
+            pipeline,
+            run_template,
+            run,
+            artifact_version_id,
+            artifact_name,
+            artifact_version,
+            infer_artifact,
+        ]
+    ):
         artifact_model = client.get_artifact(name_id_or_prefix=artifact)
         resource_id = artifact_model.id
         resource_type = TaggableResourceTypes.ARTIFACT
 
+    # Remove tags from an artifact version by its ID
+    elif artifact_version_id is not None and all(
+        v is None
+        for v in [
+            pipeline,
+            run_template,
+            run,
+            artifact,
+            artifact_name,
+            artifact_version,
+            infer_artifact,
+        ]
+    ):
+        resource_id = artifact_version_id
+        resource_type = TaggableResourceTypes.ARTIFACT_VERSION
+
     # Remove tags from an artifact version by its name and version
-    elif artifact_name is not None and artifact_version is not None:
+    elif (artifact_name is not None and artifact_version is not None) and all(
+        v is None
+        for v in [
+            pipeline,
+            run_template,
+            run,
+            artifact,
+            artifact_version_id,
+            infer_artifact,
+        ]
+    ):
         artifact_version_model = client.get_artifact_version(
             name_id_or_prefix=artifact_name, version=artifact_version
         )
         resource_id = artifact_version_model.id
         resource_type = TaggableResourceTypes.ARTIFACT_VERSION
 
-    # Remove tags from an artifact version by its ID
-    elif artifact_version_id is not None:
-        resource_id = artifact_version_id
-        resource_type = TaggableResourceTypes.ARTIFACT_VERSION
-
     # Remove tags from an artifact version through the step context
-    elif infer_artifact is True:
+    elif infer_artifact is True and all(
+        v is None
+        for v in [
+            pipeline,
+            run_template,
+            run,
+            artifact,
+            artifact_version_id,
+            artifact_version,
+        ]
+    ):
         try:
             from zenml.steps.step_context import get_step_context
 
@@ -579,12 +731,14 @@ def remove_tags(
     elif all(
         v is None
         for v in [
+            pipeline,
             run,
+            run_template,
+            artifact,
             artifact_version_id,
             artifact_name,
             artifact_version,
-            pipeline,
-            run_template,
+            infer_artifact,
         ]
     ):
         try:

zenml/zen_stores/rest_zen_store.py

@@ -4454,7 +4454,7 @@ class RestZenStore(BaseZenStore):
                         f"`zenml login {self.url}` to "
                         "re-authenticate to the server or authenticate using "
                         "an API key. See "
-                        "https://docs.zenml.io/how-to/project-setup-and-management/connecting-to-zenml/connect-with-a-service-account "
+                        "https://docs.zenml.io/deploying-zenml/connecting-to-zenml/connect-with-a-service-account "
                         "for more information."
                     )
                     # Clear the current token from the credentials store to

zenml/zen_stores/sql_zen_store.py

@@ -3001,10 +3001,6 @@ class SqlZenStore(BaseZenStore):
                     )
                     session.add(vis_schema)
 
-            # Commit the visualizations so potential future rollbacks don't
-            # remove them
-            session.commit()
-
             # Save tags of the artifact
             self._attach_tags_to_resources(
                 tags=artifact_version.tags,
@@ -5188,9 +5184,19 @@ class SqlZenStore(BaseZenStore):
                             # we want to display them as separate nodes in the
                             # DAG. We can therefore always create a new node
                             # here.
+                            is_manual_load = (
+                                input.type == StepRunInputArtifactType.MANUAL
+                            )
                             artifact_node = helper.add_artifact_node(
                                 node_id=helper.get_artifact_node_id(
-                                    name=input.name,
+                                    # For manual loads, the name might not be
+                                    # unique, so we use the artifact ID instead.
+                                    # We don't need to keep the name consistent
+                                    # with placeholder nodes as they don't exist
+                                    # for manual loads.
+                                    name=str(input.artifact_id)
+                                    if is_manual_load
+                                    else input.name,
                                     step_name=step_name,
                                     io_type=input.type,
                                     is_input=True,
@@ -5221,9 +5227,20 @@ class SqlZenStore(BaseZenStore):
                         # want to merge these and instead display them
                         # separately in the DAG, but if that should ever change
                         # this would be the place to merge them.
+                        is_manual_save = (
+                            output.artifact_version.save_type
+                            == ArtifactSaveType.MANUAL
+                        )
                         artifact_node = helper.add_artifact_node(
                             node_id=helper.get_artifact_node_id(
-                                name=output.name,
+                                # For manual saves, the name might not be
+                                # unique, so we use the artifact ID instead.
+                                # We don't need to keep the name consistent
+                                # with placeholder nodes as they don't exist
+                                # for manual saves.
+                                name=str(output.artifact_id)
+                                if is_manual_save
+                                else output.name,
                                 step_name=step_name,
                                 io_type=output.artifact_version.save_type,
                                 is_input=False,
@@ -12003,17 +12020,16 @@ class SqlZenStore(BaseZenStore):
         validate_name(tag)
         self._set_request_user_id(request_model=tag, session=session)
 
-        with session.begin_nested() as nested_session:
-            tag_schema = TagSchema.from_request(tag)
-            session.add(tag_schema)
+        tag_schema = TagSchema.from_request(tag)
+        session.add(tag_schema)
 
-            try:
-                session.commit()
-            except IntegrityError:
-                nested_session.rollback()
-                raise EntityExistsError(
-                    f"Tag with name `{tag.name}` already exists."
-                )
+        try:
+            session.commit()
+        except IntegrityError:
+            session.rollback()
+            raise EntityExistsError(
+                f"Tag with name `{tag.name}` already exists."
+            )
         return tag_schema
 
     @track_decorator(AnalyticsEvent.CREATED_TAG)
@@ -12394,6 +12410,7 @@ class SqlZenStore(BaseZenStore):
                             other_runs_with_same_tag = self.list_runs(
                                 PipelineRunFilter(
                                     id=f"notequals:{resource.id}",
+                                    project=resource.project.id,
                                     pipeline_id=resource.pipeline_id,
                                     tags=[tag_schema.name],
                                 )
@@ -12418,6 +12435,7 @@ class SqlZenStore(BaseZenStore):
                                 self.list_artifact_versions(
                                     ArtifactVersionFilter(
                                         id=f"notequals:{resource.id}",
+                                        project=resource.project.id,
                                         artifact_id=resource.artifact_id,
                                         tags=[tag_schema.name],
                                     )
@@ -12447,6 +12465,7 @@ class SqlZenStore(BaseZenStore):
                                 older_templates = self.list_run_templates(
                                     RunTemplateFilter(
                                         id=f"notequals:{resource.id}",
+                                        project=resource.project.id,
                                         pipeline_id=scope_id,
                                         tags=[tag_schema.name],
                                     )

{zenml_nightly-0.83.0.dev20250611.dist-info → zenml_nightly-0.83.0.dev20250613.dist-info}/METADATA

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

{zenml_nightly-0.83.0.dev20250611.dist-info → zenml_nightly-0.83.0.dev20250613.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=AmJkgThV_NCzJghsZ5bR4XF2rGm2az4fszG8PZix2gE,19
+zenml/VERSION,sha256=raYTX7L20222A6BNt7Djwt8Q0R7IUDK_SDF-BODdqRg,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,12 +25,12 @@ 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=44wDliyVDmfNQSRpdUwI4JhhsmGP18FF4mWgyEmN0E8,35933
-zenml/cli/__init__.py,sha256=nxq4ifwLV5qT7Qghb42h02XUOcOihBkZp2xBqgiykM8,75670
+zenml/artifacts/utils.py,sha256=_J5glTW-tkc_49sWZVhYZpDXOHR5fzKvngeJMLDTpuQ,35947
+zenml/cli/__init__.py,sha256=Q-DTKGMxUCyN8aAchAvjoWJeZ3f55Hnt4FpagiNHSbU,75655
 zenml/cli/annotator.py,sha256=JRR7_TJOWKyiKGv1kwSjG1Ay6RBWPVgm0X-D0uSBlyE,6976
 zenml/cli/artifact.py,sha256=7lsAS52DroBTFkFWxkyb-lIDOGP5jPL_Se_RDG_2jgg,9564
 zenml/cli/authorized_device.py,sha256=_1PzE3BM2SmwtuzRliEMStvbBRKWQmg_lbwCRtn8dBg,4324
-zenml/cli/base.py,sha256=9E4_9LeAAlDAusrg5-lpM4Ne1kYLkKOmUlcmhBilzaU,28373
+zenml/cli/base.py,sha256=DZfUI61nmWLR5OYZRyYqOLuBYu0UrFynLXaLVMEavE4,28372
 zenml/cli/cli.py,sha256=Pnq468IZ4oqzluA_gZ5PsrdnSPEyHcasIH-xI1_8Y_Q,5454
 zenml/cli/code_repository.py,sha256=6T3Hgv0vxNGhZ4Lb5TDw8t0Ihzv0qQS6ojFoflQ2de8,9446
 zenml/cli/config.py,sha256=UI_j0a_zRgEUd2q0zuOi4UgbjiCYjMJ_Y9iSg-wi8Oo,2768
@@ -45,17 +45,17 @@ zenml/cli/pipeline.py,sha256=Vlz1OgGb1Ep-4Ekgd-Wz5SmieWigfx56i8wA5BGl228,19222
 zenml/cli/project.py,sha256=oo9rxjwcX9Mi-0ZtiTMOoajQ8JrEkl23BcNFZxjfhj0,6536
 zenml/cli/secret.py,sha256=zwt07v0xoIf_dLf-qY5CFdbKepBEuwmXD2HIMcLe_xU,20164
 zenml/cli/served_model.py,sha256=3w1UcAbg6Geu37fr7ej1_81GBCt3fF7j3Ge799YE4Mc,14974
-zenml/cli/server.py,sha256=G4O1py2pYe4X3-WhIneAkGM6YHJXeh9EvZU_42_bGt4,26362
+zenml/cli/server.py,sha256=j5uu1Ej6v9K-yBnguGdg8-BedLxUtKaEeW20X6nLjGc,26351
 zenml/cli/service_accounts.py,sha256=DkSrBroYvQ3zNDHVYbY9IxqmbN7f8wGsAmt0Wyjq1FE,17766
 zenml/cli/service_connectors.py,sha256=ZGo1X3UwoHZ2BEFVev216Vfaf9wBEb0PKduwtX3mnk4,74332
-zenml/cli/stack.py,sha256=J5FzZWOrn9RLoJ6WjsxDl75pcTrBMiPOJeQj_lNix6U,67236
+zenml/cli/stack.py,sha256=FgO2Qx_9r_ok5FoutTbXRm6pzUp8mlaTguZQKw3YbLQ,67236
 zenml/cli/stack_components.py,sha256=QqakqWsvzgG7nOvQmvlbcfwDZUTsXI2pOlsOBXP5EXg,52963
 zenml/cli/tag.py,sha256=JiAoYyDDBBYj0ChT8zWY8KGjzAnJTNx-WLQrjLLEyik,4802
 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=cf4sYrVCx9xbX3ZcQKF2j-jRroCxy-MKUNkEOhZD1yk,293336
+zenml/client.py,sha256=YXGUxFdpxLHcl84Oe4Mc1QoXph8aBsUPUT6wOcjBEWQ,293398
 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
@@ -65,7 +65,7 @@ zenml/code_repositories/local_repository_context.py,sha256=1VyiYkJBDVg0iGusgRQDT
 zenml/config/__init__.py,sha256=DZEic7euSbwI9Yb3FMRQhTgfhqz-C6OdAiYmOb0-opI,1519
 zenml/config/base_settings.py,sha256=itoLqc1cOwEYhgSGdZmSKSaBevQkvYH7NQh7PUamazc,1700
 zenml/config/build_configuration.py,sha256=jGGNwP0Cb7a80JXArNxDgxzxl9ytSZRtv-WW7_psLbM,6870
-zenml/config/compiler.py,sha256=bK3LCDkrFc9SapJYH-vuQZ_o8scHNs-FdC5DblIUU4U,23024
+zenml/config/compiler.py,sha256=dVQ2FfliNxt93H2SYwlSOVyh-5dmdMd7abbIVRuIn3I,23023
 zenml/config/constants.py,sha256=QvSgMwXWxtspcJ45CrFDP1ZY3w6gS3bIhXLOtIDAbZA,713
 zenml/config/docker_settings.py,sha256=xZvoeC6v6RG_5MFK_RgdTgrkOLu-QT6frY4SGQNZUTo,18460
 zenml/config/global_config.py,sha256=59orl6xp4moxSo2Q8tXtDfpIBX4AxJ9FHcm8s2VHyWw,29712
@@ -75,7 +75,7 @@ zenml/config/pipeline_spec.py,sha256=uWpiIfznJvXAiKs1yMIUDT1h1tYEFNO-RDVTYcIv9CE
 zenml/config/resource_settings.py,sha256=0taXGHvDfXuvpMplxsuzpznB1sAvWJIGnXoVJ9-ySfw,3899
 zenml/config/retry_config.py,sha256=4UH1xqw0G6fSEbXSNKfmiFEkwadxQef9BGMe3JBm6NI,929
 zenml/config/schedule.py,sha256=-83j99U9OyiG7E322XWA7QvuLSwQzF21whwpeiF0b30,5348
-zenml/config/secret_reference_mixin.py,sha256=YvY68MTd1gE23IVprf0BLkNn62hoxcvb5nqGgc8jMkU,5871
+zenml/config/secret_reference_mixin.py,sha256=SKmvPkoe78V8wM1gQMGS0e2VmvwjMAwG2E474fLqs9Q,5871
 zenml/config/secrets_store_config.py,sha256=y05zqyQhr_DGrs3IfBGa_FRoZ043hSYRT5wzrx-zHTU,2818
 zenml/config/server_config.py,sha256=DYYQ10HddkvlCWobBpwpg8ggO6imMZCnfFxzqlkQg_U,32336
 zenml/config/settings_resolver.py,sha256=PR9BRm_x1dy7nVKa9UqpeFdck8IEATSW6aWT8FKd-DI,4278
@@ -141,12 +141,12 @@ zenml/integrations/aws/container_registries/aws_container_registry.py,sha256=SCx
 zenml/integrations/aws/flavors/__init__.py,sha256=XYL9fpwKzeFfHCjakU0iJ3SyHVRJk63QT7luOy9Giek,1480
 zenml/integrations/aws/flavors/aws_container_registry_flavor.py,sha256=GIDLOySz1zF08YSkmKIj89TxBqSLWueXLAHyxYwm-NI,4169
 zenml/integrations/aws/flavors/aws_image_builder_flavor.py,sha256=XobJOw5ymbY22i7YHWGYOKDQoJQAqTeMIfvkADt-DDc,5343
-zenml/integrations/aws/flavors/sagemaker_orchestrator_flavor.py,sha256=aPBuzQjkroSZOdGWynXYu_A1c0YjK5dB96zb0NvHkBs,13604
+zenml/integrations/aws/flavors/sagemaker_orchestrator_flavor.py,sha256=rcMCoYMFxL8AtwMUf_wdnmQrIMmBwryNDfoU39pVC2E,13611
 zenml/integrations/aws/flavors/sagemaker_step_operator_flavor.py,sha256=e3locb2OnF7bqV3iafIUB_tHhDE8-i7eyB4H6Hyqj1Y,6084
 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=EfDFT2WG0t-6x1CU0n57BS7YCC0wPA8CfTGpb4-G0Tc,39282
+zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py,sha256=EtZpTgK-Z9Q4Eb7cM6VGXnrAkiJjpFrIdtQkHiw0Hhw,39289
 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
@@ -269,7 +269,7 @@ zenml/integrations/gcp/google_credentials_mixin.py,sha256=bPy3JYCCcyuTmPiVFqbY81
 zenml/integrations/gcp/image_builders/__init__.py,sha256=2IvTL6U2YpUoxGQXeXew-6WFoL5hHIxkqr4DaA5Ez9w,786
 zenml/integrations/gcp/image_builders/gcp_image_builder.py,sha256=5T6BXsHxLhvp1BF_rslXl1oZzykJUPuZ3E_7-9ZZYLk,9019
 zenml/integrations/gcp/orchestrators/__init__.py,sha256=6xLFJKZKQk73fHPF-XdpbQO87zjQNGTsNHjJjLfG_Kg,805
-zenml/integrations/gcp/orchestrators/vertex_orchestrator.py,sha256=qoMCr36buZUz0y4CyTFQde3RDslkaGLAG0FjXc0XEPU,42100
+zenml/integrations/gcp/orchestrators/vertex_orchestrator.py,sha256=r2q-iHPFm6sQKcKitCvuYqfMJz6mPXQ0WvWvFYDFmto,42100
 zenml/integrations/gcp/service_connectors/__init__.py,sha256=fdydawaor8KAtMYvRZieiTuA1i5QATxXXgI-yV1lsn8,788
 zenml/integrations/gcp/service_connectors/gcp_service_connector.py,sha256=9u-vEHbmSyN5IGwYI8v39TcFZg5ObgkxlbwSPz-e5zE,95018
 zenml/integrations/gcp/step_operators/__init__.py,sha256=iPkob2LtPIQ-OHszhbNz_ojhoovL6SprmTx37It4EJ8,808
@@ -404,7 +404,7 @@ zenml/integrations/modal/step_operators/modal_step_operator.py,sha256=J2HiNaGnlP
 zenml/integrations/neptune/__init__.py,sha256=PcqYVwybBcEy5YFODNA6bxT91kxaETReFmCWzExGhM8,1627
 zenml/integrations/neptune/experiment_trackers/__init__.py,sha256=DJi7lk7NXYrRg3VGPPSEsMycKECDfXL-h2V0A0r7z8Y,833
 zenml/integrations/neptune/experiment_trackers/neptune_experiment_tracker.py,sha256=hqEOgyBEQhjcnxvgw2oC2WZ0FAv1ez4VAqa_fJAHeCw,3716
-zenml/integrations/neptune/experiment_trackers/run_state.py,sha256=W-3mnjoVT-lZNVxGm4982jvMYbxC9WNiQ9_0vd4f9SM,5411
+zenml/integrations/neptune/experiment_trackers/run_state.py,sha256=Qvge0eaUs70NQIBpsY5P8vRHHS1hHCAlZkZsmiGgoqk,5418
 zenml/integrations/neptune/flavors/__init__.py,sha256=NzORpmtI3Vu7yH1maj5pYd_2gQoN4QR2ZZLy1uPr6uw,975
 zenml/integrations/neptune/flavors/neptune_experiment_tracker_flavor.py,sha256=dGlRmnPjG5qG1RlcpEmpANDKAigh1N4hE86jvmaC968,3634
 zenml/integrations/neptune/neptune_constants.py,sha256=-VddhrALS1HMBhGtFiGDKaRahC-qhG0JAF2zAuc3lMM,746
@@ -596,9 +596,9 @@ zenml/login/pro/workspace/models.py,sha256=godIY1q2dUxB0QILaOFOeqV1I3WglWUjZC11n
 zenml/login/server_info.py,sha256=-_sK2L-curHdzUv1JDOwF6GoEeAXT5vFZN0J-5Ug4wU,1663
 zenml/login/web_login.py,sha256=y0vDwonQ-8wMjqMeRW0ANLoZR9X5OKmz-8qCzCiuBZo,9175
 zenml/materializers/__init__.py,sha256=C3lZaTmIFxwIPwCKF8oLQUkLaX2o7Dbj9hvYVFrSzt8,1758
-zenml/materializers/base_materializer.py,sha256=M4hwkw7PB0LskCE92r-S35011l7DlFemit-EuUCW3Nc,14002
+zenml/materializers/base_materializer.py,sha256=wMRLOAz0arRU2Q6tCxcYVfT62I-qg7HVHOPUd1OtKAg,13939
 zenml/materializers/built_in_materializer.py,sha256=HgcCHg3GpuQ4t-jecYcdg0kldUfLeUJvIpm8-wcS11I,17189
-zenml/materializers/cloudpickle_materializer.py,sha256=x8a6jEMTky6N2YVHiwrnGWSfVJUpiy-4kQsD2Aqj_E0,4837
+zenml/materializers/cloudpickle_materializer.py,sha256=PIZauXg1SBYmBUi9FBd2t3LFyZupp3jYtMjZ0OTug18,4816
 zenml/materializers/materializer_registry.py,sha256=ic-aWhJ2Ex9F_rml2dDVAxhRfW3nd71QMxzfTPP6BIM,4002
 zenml/materializers/numpy_materializer.py,sha256=OLcHF9Z0tAqQ_U8TraA0vGmZjHoT7eT_XevncIutt0M,1715
 zenml/materializers/pandas_materializer.py,sha256=c4B-ly04504gysA66iCYcmEdeh0ClePRTxRCkmHqIgE,1725
@@ -613,7 +613,7 @@ zenml/metadata/metadata_types.py,sha256=ts1EhF2qGZb8siKv1nkPSeFeyR2kbiIODkpk-hyo
 zenml/model/__init__.py,sha256=bFPHnWCgAGAjUPCmODHUmwbB0KGljNSEik857Yi-QX0,673
 zenml/model/lazy_load.py,sha256=nnu37QaIPU0peqVCEwG3k37LJe_D1i6RCs_8xoId6yk,4583
 zenml/model/model.py,sha256=i8xcvD0nxQfmi1O5dctZGQXE8SVtmmK1If1OaYKcAwA,27231
-zenml/model/utils.py,sha256=R7OuuC-wI5331reaqSg8AkpfRNIPSWzGHfL6w9Eyadc,6013
+zenml/model/utils.py,sha256=-NNG1xALQWHyNNJu9AMfr5WejdRe0_R-JngxsW45QvU,5978
 zenml/model_deployers/__init__.py,sha256=oVBLtTfrNenl5OI1iqtQUvJ0vpocRVUN_HIt8qpoZmY,1730
 zenml/model_deployers/base_model_deployer.py,sha256=Z5-E_zC0Ss5OQhUBz9YKeVgS2MJOzuNz2aBzqI5eo4U,24610
 zenml/model_registries/__init__.py,sha256=wA9Vzo0w_e9zuXOVURB9w8oMLSnTaimXcxg_Nb7O3b0,1238
@@ -681,7 +681,7 @@ zenml/models/v2/misc/statistics.py,sha256=ajce9rHC2bBylLzOmLfcOBSbTnJD67Y8n5YKCY
 zenml/models/v2/misc/tag.py,sha256=jUoz7wqMpDFlIUmvj4PVq8NYJJB7TMCcdRfW-DAABCU,974
 zenml/models/v2/misc/user_auth.py,sha256=1-yafNA9qK4wL8ToROjaklTVI7Mj9va0t80_4wm7w3U,6988
 zenml/orchestrators/__init__.py,sha256=p4Y9Ni7Lp33fZ6UrjgI7qu-rrg8LrlyafCM-K1WC81w,1961
-zenml/orchestrators/base_orchestrator.py,sha256=EcxtHluCl8ov37hFk-7ncrCiozJJX34yl_eg44HcjxM,12127
+zenml/orchestrators/base_orchestrator.py,sha256=cjD-2OSrvh6ZuZ8ehM7rjg9RhkD_WaALgYlBJ67q5bY,12096
 zenml/orchestrators/cache_utils.py,sha256=QkmTs-ANfXve9_QzTqgGlyulZDEWOngoTcsiSjG5aA8,5906
 zenml/orchestrators/containerized_orchestrator.py,sha256=rdebgBW0Bk--JcHcT0NpLkAbyhY0VS5xO1uwWEgkLpA,3230
 zenml/orchestrators/dag_runner.py,sha256=4oAGc52nHv8HeI_Vj7GH1jzjJom1uwUJ_la9usQoOFY,9404
@@ -696,13 +696,13 @@ zenml/orchestrators/step_launcher.py,sha256=6hrLj0Dr5-FcKTm3cvLVnr3PuUpaYbyEs1MR
 zenml/orchestrators/step_run_utils.py,sha256=FX7msn7yov-AMbtrdpoY_fst1ge65GoRFIIVdxuVyds,14898
 zenml/orchestrators/step_runner.py,sha256=EUgKG_g0fOQ6gnB1hPSSa6UXwUKVkguC-Yj-Q0yEQXg,26632
 zenml/orchestrators/topsort.py,sha256=D8evz3X47zwpXd90NMLsJD-_uCeXtV6ClzNfDUrq7cM,5784
-zenml/orchestrators/utils.py,sha256=agh8m5v1OYU-16Rcp6YDwDbHt9S1F46zsRe3AWC_tBc,12967
+zenml/orchestrators/utils.py,sha256=6bqLc1fmdJTXg8JUwUKs8YNbmxTuMIfWmUbUpg-7hx0,12956
 zenml/orchestrators/wheeled_orchestrator.py,sha256=eOnMcnd3sCzfhA2l6qRAzF0rOXzaojbjvvYvTkqixQo,4791
 zenml/pipelines/__init__.py,sha256=hpIX7hN8jsQRHT5R-xSXZL88qrHwkmrvGLQeu1rWt4o,873
 zenml/pipelines/build_utils.py,sha256=DltGesybT8qYum4i23mvWZlVRgp7UxWdbHd1Y9ySv5c,27889
 zenml/pipelines/pipeline_context.py,sha256=4BixReLcPo33VtNBDrMwnJqjKTinHjmO5AOfmoeIOQM,3659
 zenml/pipelines/pipeline_decorator.py,sha256=LB21QYrbFeBdUGwKBUNbdpXAxO4OOtYl5Vs_mzJNXqU,4600
-zenml/pipelines/pipeline_definition.py,sha256=V_5IQH93RKtv9vUblJcYCuTs6HydgDEOcqTGPdMQrTM,59637
+zenml/pipelines/pipeline_definition.py,sha256=CVZDquIjq8WaiRwJuckTH_SxLnkR08g1FdZxCsJlcUU,59634
 zenml/pipelines/run_utils.py,sha256=VouyPLQm9QgI2zZuc-F8XRW_beb1Uew4vt6ucT9bC-E,12288
 zenml/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zenml/plugins/base_plugin_flavor.py,sha256=88IxFW91UB_rQ8xPlfRnIhIJh7A308NEq2epMMdlOng,2530
@@ -719,7 +719,7 @@ zenml/service_connectors/__init__.py,sha256=gIBAVUPBZtdO6JtEBCixy9YG4wZRA1mnaxaG
 zenml/service_connectors/docker_service_connector.py,sha256=_6WPqYCcSUf8JPakipbkJRvaN2ghhY7zCr38WCHG0SI,13218
 zenml/service_connectors/service_connector.py,sha256=8Z_S_1qcfTla9r-Yu2hOmWaVYQ8C9jbkTSawu5QzY6o,55139
 zenml/service_connectors/service_connector_registry.py,sha256=U5MdC9M_RfkHK56AoERB9HtDP4vxk2hDFoZRlWzBA9g,9608
-zenml/service_connectors/service_connector_utils.py,sha256=25eD-2BDWHmRFwpdV5xCAtM3osgGwBzZcZecYRox6Tk,17958
+zenml/service_connectors/service_connector_utils.py,sha256=jcJIClAVHPasqPy1QKqrzSmW1UHcHXekaX4cT-5exUU,17937
 zenml/services/__init__.py,sha256=BhqBztNXLrqN-4FZr24WHP9ujP_M0plt6axQvakh1jQ,2784
 zenml/services/container/__init__.py,sha256=dFHcmlXgNXjUaCR18oGQJ19-I_6f3UeAUURHjqldjTs,668
 zenml/services/container/container_service.py,sha256=EJ8RM0llzCwKxVZj3Oyhu7HC0K8XNWP-VQhIutff-kw,18403
@@ -738,7 +738,7 @@ zenml/stack/authentication_mixin.py,sha256=_Rn6SurHnyBuU_CJBOkwPR305yQFfqN60t6nE
 zenml/stack/flavor.py,sha256=wFLjajCw_G2NMR2UJzEGH2dcLj1dvmU9dDI5iS8b9rk,10544
 zenml/stack/flavor_registry.py,sha256=IL0fRrxxQJ9YkCYCeADP7nwWEQo4XBElJY4owMjKGbQ,6108
 zenml/stack/stack.py,sha256=3mLGNBuep0PeBcoFfOEaB6Ya206yav1ppiH5YEjp9xA,33042
-zenml/stack/stack_component.py,sha256=eshFuuknjtA5yT1d8aWPoWMkDph-K8E06eeZ-h__T38,29237
+zenml/stack/stack_component.py,sha256=QTPqU48xLVNRIocumzQQrnE-wwgZab9dMkPzOTIo3lc,29237
 zenml/stack/stack_validator.py,sha256=hWbvvGIeWLj6NwSsF4GCc6RAxAWvxHXTcBZL9nJvcak,3111
 zenml/stack/utils.py,sha256=CHs9rxdqHkUT12KPhJX1YPtIWnZBoVlRlq5PzNymq3E,6406
 zenml/stack_deployments/__init__.py,sha256=-7593cQ_ZgRn774Ol-8AKXXQquIU4DSiaThVEr6TfWM,644
@@ -754,7 +754,7 @@ zenml/step_operators/step_operator_entrypoint_configuration.py,sha256=WoNO-fXukV
 zenml/steps/__init__.py,sha256=KKWFOmCZGLDEikOD2E5YmDA7QHo47uPV37by21WwI0U,1453
 zenml/steps/base_step.py,sha256=qzW9I99PWK8brm3wdV1NEuUxGFlMvrmqNh3YjNKUkgg,44312
 zenml/steps/decorated_step.py,sha256=C8Ng5PCLc9eql4JF1N345HQ6LyC1qCUdTnysUTeoAJs,1315
-zenml/steps/entrypoint_function_utils.py,sha256=AuBIKqcquuvSTNhvuavtSuPdzzEfdOa3_h3mt7twfrA,9577
+zenml/steps/entrypoint_function_utils.py,sha256=H0WIkHpR_R0S9gl_tWr0nX-fcBlYnm8OQ1cimvrw-qo,9555
 zenml/steps/step_context.py,sha256=sFvVVvEyKmWTrucofor8Cuxv-72jbziVaQY-OlOvFAM,15526
 zenml/steps/step_decorator.py,sha256=cbu7s2EZ6-yIgBlY6npour2X_CnUCkBfcXfJoiSb3iQ,6454
 zenml/steps/step_invocation.py,sha256=ETfOaV-P4_iXGk9y1-xK54Kfg2QRYaGoj_jTyEYZfb0,4861
@@ -797,7 +797,7 @@ zenml/utils/singleton.py,sha256=uFRrUlUdS5VyY9lLJyl_n5kqppsqJLKkBhSj4g5VPkY,2757
 zenml/utils/source_code_utils.py,sha256=8iyNA2MGIORYVEkSdxNTXfS1ZdFKXTAG1dZRkeQtPL0,3751
 zenml/utils/source_utils.py,sha256=joKLghhDq9dh0fd8B0WRGX-nN-uwnGQdgmsyY_n-8gY,27033
 zenml/utils/string_utils.py,sha256=xJ8Abm52yFQyOpNrgpoLjDbPCgb6rpJsi8N4-7bb5GU,7254
-zenml/utils/tag_utils.py,sha256=2LJ-3XkXuWnffUsyfCwahzL_N2k4lIYIgBs1HVShhA0,19180
+zenml/utils/tag_utils.py,sha256=GjuTyrkZAPhT27nP41bttBAvqGFpzgVZ9WZ_mswLc6o,22543
 zenml/utils/time_utils.py,sha256=-9Y9zwJ-6Gv7hoZQCoftPyC2LCLo2bYj6OgdyBaE44o,4076
 zenml/utils/typed_model.py,sha256=00EAo1I1VnOBHG4-ce8dPkyHRPpgi67SRIU-KdewRWs,4757
 zenml/utils/typing_utils.py,sha256=jP7JKrlLsuMIStXhwKFNWykE6SMOR72tJIJ_qEbQSNc,6555
@@ -1281,7 +1281,7 @@ zenml/zen_stores/migrations/versions/f49904a80aa7_increase_length_of_artifact_ta
 zenml/zen_stores/migrations/versions/f76a368a25a5_add_stack_description.py,sha256=u8fRomaasFeGhxvM2zU-Ab-AEpVsWm5zRcixxKFXdRw,904
 zenml/zen_stores/migrations/versions/fbd7f18ced1e_increase_step_run_field_lengths.py,sha256=kn-ng5EHe_mmLfffIFbz7T59z-to3oMx8III_4wOsz4,1956
 zenml/zen_stores/migrations/versions/ff538a321a92_migrate_onboarding_state.py,sha256=gsUFLJQ32_o9U35JCVqkqJVVk-zfq3yel25hXhzVFm4,3829
-zenml/zen_stores/rest_zen_store.py,sha256=1n2g6qR4TdvskFXOPdR_rLLhLOYgeVYZfNOwiBXz-DE,160139
+zenml/zen_stores/rest_zen_store.py,sha256=S7eMMfd7unRX1MzvTLaHNQ_GoUOni4izizfJdgyTXsw,160119
 zenml/zen_stores/schemas/__init__.py,sha256=4EXqExiVyxdnGxhQ_Hz79mOdRuMD0LsGlw0PaP2Ef6o,4333
 zenml/zen_stores/schemas/action_schemas.py,sha256=jymEro5r20seTpRcAr5iZw-g8rXZRSZFEbiEm_iFaj8,7618
 zenml/zen_stores/schemas/api_key_schemas.py,sha256=RJS-lH7AHHAe9oXA0CGWvtbrmYgVfNP9nMa7rhtRNnY,8379
@@ -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=zgdaIUTG2JYTr6_sW_yhYar_ZgPQZiKWEO3rmM0ys1Y,466071
+zenml/zen_stores/sql_zen_store.py,sha256=uYYwCvz-ot-qHPModf7Z6FheVRtZj7TUiWw-tHaBzGI,467346
 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.dev20250611.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.83.0.dev20250611.dist-info/METADATA,sha256=n7QUbanOkHx11TvKqo-2zfDF75fepC5lxY3zn3mBPm8,24317
-zenml_nightly-0.83.0.dev20250611.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.83.0.dev20250611.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.83.0.dev20250611.dist-info/RECORD,,
+zenml_nightly-0.83.0.dev20250613.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.83.0.dev20250613.dist-info/METADATA,sha256=2iedUCzK58EUDGXITPNywLGmCuqlhXKHvyfj8d7JE7A,24317
+zenml_nightly-0.83.0.dev20250613.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.83.0.dev20250613.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.83.0.dev20250613.dist-info/RECORD,,