zenml-nightly 0.75.0.dev20250318__py3-none-any.whl → 0.80.0.dev20250322__py3-none-any.whl

zenml/models/v2/core/pipeline_run.py

@@ -42,6 +42,7 @@ from zenml.models.v2.base.scoped import (
     ProjectScopedResponseBody,
     ProjectScopedResponseMetadata,
     ProjectScopedResponseResources,
+    RunMetadataFilterMixin,
     TaggableFilter,
 )
 from zenml.models.v2.core.model_version import ModelVersionResponse
@@ -582,12 +583,15 @@ class PipelineRunResponse(
 # ------------------ Filter Model ------------------
 
 
-class PipelineRunFilter(ProjectScopedFilter, TaggableFilter):
+class PipelineRunFilter(
+    ProjectScopedFilter, TaggableFilter, RunMetadataFilterMixin
+):
     """Model to enable advanced filtering of all pipeline runs."""
 
     CUSTOM_SORTING_OPTIONS: ClassVar[List[str]] = [
         *ProjectScopedFilter.CUSTOM_SORTING_OPTIONS,
         *TaggableFilter.CUSTOM_SORTING_OPTIONS,
+        *RunMetadataFilterMixin.CUSTOM_SORTING_OPTIONS,
         "tag",
         "stack",
         "pipeline",
@@ -597,6 +601,7 @@ class PipelineRunFilter(ProjectScopedFilter, TaggableFilter):
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
         *ProjectScopedFilter.FILTER_EXCLUDE_FIELDS,
         *TaggableFilter.FILTER_EXCLUDE_FIELDS,
+        *RunMetadataFilterMixin.FILTER_EXCLUDE_FIELDS,
         "unlisted",
         "code_repository_id",
         "build_id",
@@ -610,11 +615,16 @@ class PipelineRunFilter(ProjectScopedFilter, TaggableFilter):
         "stack_component",
         "pipeline_name",
         "templatable",
-        "run_metadata",
     ]
     CLI_EXCLUDE_FIELDS = [
         *ProjectScopedFilter.CLI_EXCLUDE_FIELDS,
         *TaggableFilter.CLI_EXCLUDE_FIELDS,
+        *RunMetadataFilterMixin.CLI_EXCLUDE_FIELDS,
+    ]
+    API_MULTI_INPUT_PARAMS: ClassVar[List[str]] = [
+        *ProjectScopedFilter.API_MULTI_INPUT_PARAMS,
+        *TaggableFilter.API_MULTI_INPUT_PARAMS,
+        *RunMetadataFilterMixin.API_MULTI_INPUT_PARAMS,
     ]
 
     name: Optional[str] = Field(
@@ -680,10 +690,6 @@ class PipelineRunFilter(ProjectScopedFilter, TaggableFilter):
         union_mode="left_to_right",
     )
     unlisted: Optional[bool] = None
-    run_metadata: Optional[Dict[str, Any]] = Field(
-        default=None,
-        description="The run_metadata to filter the pipeline runs by.",
-    )
     # TODO: Remove once frontend is ready for it. This is replaced by the more
     #   generic `pipeline` filter below.
     pipeline_name: Optional[str] = Field(
@@ -740,8 +746,6 @@ class PipelineRunFilter(ProjectScopedFilter, TaggableFilter):
             PipelineDeploymentSchema,
             PipelineRunSchema,
             PipelineSchema,
-            RunMetadataResourceSchema,
-            RunMetadataSchema,
             ScheduleSchema,
             StackComponentSchema,
             StackCompositionSchema,
@@ -899,30 +903,6 @@ class PipelineRunFilter(ProjectScopedFilter, TaggableFilter):
                 )
 
             custom_filters.append(templatable_filter)
-        if self.run_metadata is not None:
-            from zenml.enums import MetadataResourceTypes
-
-            for key, value in self.run_metadata.items():
-                additional_filter = and_(
-                    RunMetadataResourceSchema.resource_id
-                    == PipelineRunSchema.id,
-                    RunMetadataResourceSchema.resource_type
-                    == MetadataResourceTypes.PIPELINE_RUN.value,
-                    RunMetadataResourceSchema.run_metadata_id
-                    == RunMetadataSchema.id,
-                    self.generate_custom_query_conditions_for_column(
-                        value=key,
-                        table=RunMetadataSchema,
-                        column="key",
-                    ),
-                    self.generate_custom_query_conditions_for_column(
-                        value=value,
-                        table=RunMetadataSchema,
-                        column="value",
-                        json_encode_value=True,
-                    ),
-                )
-                custom_filters.append(additional_filter)
 
         return custom_filters
 

zenml/models/v2/core/service.py

@@ -31,6 +31,7 @@ from pydantic import ConfigDict, Field
 from sqlalchemy.sql.elements import ColumnElement
 
 from zenml.constants import STR_FIELD_MAX_LENGTH
+from zenml.enums import ServiceState
 from zenml.models.v2.base.base import BaseUpdate
 from zenml.models.v2.base.scoped import (
     ProjectScopedFilter,
@@ -40,8 +41,7 @@ from zenml.models.v2.base.scoped import (
     ProjectScopedResponseMetadata,
     ProjectScopedResponseResources,
 )
-from zenml.services.service_status import ServiceState
-from zenml.services.service_type import ServiceType
+from zenml.models.v2.misc.service import ServiceType
 
 if TYPE_CHECKING:
     from zenml.models.v2.core.model_version import ModelVersionResponse

zenml/models/v2/core/step_run.py

@@ -16,7 +16,6 @@
 from datetime import datetime
 from typing import (
     TYPE_CHECKING,
-    Any,
     ClassVar,
     Dict,
     List,
@@ -41,6 +40,7 @@ from zenml.models.v2.base.scoped import (
     ProjectScopedResponseBody,
     ProjectScopedResponseMetadata,
     ProjectScopedResponseResources,
+    RunMetadataFilterMixin,
 )
 from zenml.models.v2.core.artifact_version import ArtifactVersionResponse
 from zenml.models.v2.core.model_version import ModelVersionResponse
@@ -504,13 +504,25 @@ class StepRunResponse(
 # ------------------ Filter Model ------------------
 
 
-class StepRunFilter(ProjectScopedFilter):
+class StepRunFilter(ProjectScopedFilter, RunMetadataFilterMixin):
     """Model to enable advanced filtering of step runs."""
 
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
         *ProjectScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *RunMetadataFilterMixin.FILTER_EXCLUDE_FIELDS,
         "model",
-        "run_metadata",
+    ]
+    CLI_EXCLUDE_FIELDS: ClassVar[List[str]] = [
+        *ProjectScopedFilter.CLI_EXCLUDE_FIELDS,
+        *RunMetadataFilterMixin.CLI_EXCLUDE_FIELDS,
+    ]
+    CUSTOM_SORTING_OPTIONS: ClassVar[List[str]] = [
+        *ProjectScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *RunMetadataFilterMixin.CUSTOM_SORTING_OPTIONS,
+    ]
+    API_MULTI_INPUT_PARAMS: ClassVar[List[str]] = [
+        *ProjectScopedFilter.API_MULTI_INPUT_PARAMS,
+        *RunMetadataFilterMixin.API_MULTI_INPUT_PARAMS,
     ]
 
     name: Optional[str] = Field(
@@ -563,10 +575,6 @@ class StepRunFilter(ProjectScopedFilter):
         default=None,
         description="Name/ID of the model associated with the step run.",
     )
-    run_metadata: Optional[Dict[str, Any]] = Field(
-        default=None,
-        description="The run_metadata to filter the step runs by.",
-    )
     model_config = ConfigDict(protected_namespaces=())
 
     def get_custom_filters(
@@ -587,8 +595,6 @@ class StepRunFilter(ProjectScopedFilter):
         from zenml.zen_stores.schemas import (
             ModelSchema,
             ModelVersionSchema,
-            RunMetadataResourceSchema,
-            RunMetadataSchema,
             StepRunSchema,
         )
 
@@ -601,28 +607,5 @@ class StepRunFilter(ProjectScopedFilter):
                 ),
             )
             custom_filters.append(model_filter)
-        if self.run_metadata is not None:
-            from zenml.enums import MetadataResourceTypes
-
-            for key, value in self.run_metadata.items():
-                additional_filter = and_(
-                    RunMetadataResourceSchema.resource_id == StepRunSchema.id,
-                    RunMetadataResourceSchema.resource_type
-                    == MetadataResourceTypes.STEP_RUN.value,
-                    RunMetadataResourceSchema.run_metadata_id
-                    == RunMetadataSchema.id,
-                    self.generate_custom_query_conditions_for_column(
-                        value=key,
-                        table=RunMetadataSchema,
-                        column="key",
-                    ),
-                    self.generate_custom_query_conditions_for_column(
-                        value=value,
-                        table=RunMetadataSchema,
-                        column="value",
-                        json_encode_value=True,
-                    ),
-                )
-                custom_filters.append(additional_filter)
 
         return custom_filters

zenml/{services/service_type.py → models/v2/misc/service.py}

@@ -11,7 +11,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
-"""Implementation of a ZenML ServiceType class."""
+"""Misc models for services."""
 
 from pydantic import BaseModel, ConfigDict
 

zenml/orchestrators/step_run_utils.py

@@ -352,7 +352,7 @@ def log_model_version_dashboard_url(
     """
     from zenml.utils.dashboard_utils import get_model_version_url
 
-    if model_version_url := get_model_version_url(model_version.id):
+    if model_version_url := get_model_version_url(model_version):
         logger.info(
             "Dashboard URL for Model Version `%s (%s)`:\n%s",
             model_version.model.name,

zenml/orchestrators/utils.py

@@ -184,7 +184,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/connecting-to-zenml/connect-with-a-service-account"
+                        "https://docs.zenml.io/how-to/manage-zenml-server/connecting-to-zenml/connect-with-a-service-account"
                     )
 
                 # The schedule, pipeline run or step run credentials are scoped to

zenml/services/__init__.py

@@ -51,19 +51,17 @@ from zenml.services.service_monitor import (
     TCPEndpointHealthMonitor,
     TCPEndpointHealthMonitorConfig,
 )
-from zenml.services.service_status import ServiceState, ServiceStatus
-from zenml.services.service_type import ServiceType
-
+from zenml.services.service_status import ServiceStatus
+from zenml.enums import ServiceState
+from zenml.models.v2.misc.service import ServiceType
 
 __all__ = [
-    "ServiceState",
     "ServiceConfig",
     "ServiceStatus",
     "ServiceEndpointProtocol",
     "ServiceEndpointConfig",
     "ServiceEndpointStatus",
     "BaseServiceEndpoint",
-    "ServiceType",
     "BaseService",
     "ServiceEndpointHealthMonitorConfig",
     "BaseServiceEndpointHealthMonitor",

zenml/services/container/container_service.py

@@ -27,12 +27,13 @@ from docker.models.containers import Container
 from pydantic import Field
 
 from zenml.constants import ENV_ZENML_CONFIG_PATH
+from zenml.enums import ServiceState
 from zenml.logger import get_logger
 from zenml.services.container.container_service_endpoint import (
     ContainerServiceEndpoint,
 )
 from zenml.services.service import BaseService, ServiceConfig
-from zenml.services.service_status import ServiceState, ServiceStatus
+from zenml.services.service_status import ServiceStatus
 from zenml.utils import docker_utils
 from zenml.utils.io_utils import (
     create_dir_recursive_if_not_exists,

zenml/services/local/local_service.py

@@ -26,12 +26,13 @@ import psutil
 from psutil import NoSuchProcess
 from pydantic import Field
 
+from zenml.enums import ServiceState
 from zenml.logger import get_logger
 from zenml.services.local.local_service_endpoint import (
     LocalDaemonServiceEndpoint,
 )
 from zenml.services.service import BaseService, ServiceConfig
-from zenml.services.service_status import ServiceState, ServiceStatus
+from zenml.services.service_status import ServiceStatus
 from zenml.utils.io_utils import create_dir_recursive_if_not_exists
 
 logger = get_logger(__name__)

zenml/services/service.py

@@ -34,11 +34,12 @@ from uuid import UUID
 from pydantic import ConfigDict
 
 from zenml.console import console
+from zenml.enums import ServiceState
 from zenml.logger import get_logger
+from zenml.models.v2.misc.service import ServiceType
 from zenml.services.service_endpoint import BaseServiceEndpoint
 from zenml.services.service_monitor import HTTPEndpointHealthMonitor
-from zenml.services.service_status import ServiceState, ServiceStatus
-from zenml.services.service_type import ServiceType
+from zenml.services.service_status import ServiceStatus
 from zenml.utils import source_utils
 from zenml.utils.typed_model import BaseTypedModel
 

zenml/services/service_endpoint.py

@@ -16,9 +16,10 @@
 from typing import Any, Optional, Tuple
 
 from zenml.constants import DEFAULT_LOCAL_SERVICE_IP_ADDRESS
+from zenml.enums import ServiceState
 from zenml.logger import get_logger
 from zenml.services.service_monitor import BaseServiceEndpointHealthMonitor
-from zenml.services.service_status import ServiceState, ServiceStatus
+from zenml.services.service_status import ServiceStatus
 from zenml.utils.enum_utils import StrEnum
 from zenml.utils.typed_model import BaseTypedModel
 

zenml/services/service_monitor.py

@@ -19,8 +19,8 @@ from typing import TYPE_CHECKING, Optional, Tuple
 import requests
 from pydantic import Field
 
+from zenml.enums import ServiceState
 from zenml.logger import get_logger
-from zenml.services.service_status import ServiceState
 from zenml.utils.networking_utils import port_is_open
 from zenml.utils.typed_model import BaseTypedModel
 

zenml/services/service_status.py

@@ -15,24 +15,13 @@
 
 from typing import Optional
 
+from zenml.enums import ServiceState
 from zenml.logger import get_logger
-from zenml.utils.enum_utils import StrEnum
 from zenml.utils.typed_model import BaseTypedModel
 
 logger = get_logger(__name__)
 
 
-class ServiceState(StrEnum):
-    """Possible states for the service and service endpoint."""
-
-    INACTIVE = "inactive"
-    ACTIVE = "active"
-    PENDING_STARTUP = "pending_startup"
-    PENDING_SHUTDOWN = "pending_shutdown"
-    ERROR = "error"
-    SCALED_TO_ZERO = "scaled_to_zero"
-
-
 class ServiceStatus(BaseTypedModel):
     """Information about the status of a service or process.
 

zenml/stack_deployments/aws_stack_deployment.py

@@ -17,6 +17,10 @@ from typing import ClassVar, Dict, List, Optional
 
 from zenml.enums import StackDeploymentProvider
 from zenml.models import StackDeploymentConfig
+from zenml.stack_deployments.constants import (
+    TERRAFORM_AWS_MODULE_VERSION_SPEC,
+    TERRAFORM_PROVIDER_VERSION_SPEC,
+)
 from zenml.stack_deployments.stack_deployment import (
     STACK_DEPLOYMENT_TERRAFORM,
     ZenMLCloudStackDeployment,
@@ -203,10 +207,6 @@ console.
         # Based on the AWS regions listed at
         # https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/using-regions-availability-zones.html
         return {
-            "US East (Ohio)": "us-east-2",
-            "US East (N. Virginia)": "us-east-1",
-            "US West (N. California)": "us-west-1",
-            "US West (Oregon)": "us-west-2",
             "Africa (Cape Town)": "af-south-1",
             "Asia Pacific (Hong Kong)": "ap-east-1",
             "Asia Pacific (Hyderabad)": "ap-south-2",
@@ -232,6 +232,10 @@ console.
             "Middle East (Bahrain)": "me-south-1",
             "Middle East (UAE)": "me-central-1",
             "South America (São Paulo)": "sa-east-1",
+            "US East (Ohio)": "us-east-2",
+            "US East (N. Virginia)": "us-east-1",
+            "US West (N. California)": "us-west-1",
+            "US West (Oregon)": "us-west-2",
         }
 
     def get_deployment_config(
@@ -287,6 +291,7 @@ console.
         }}
         zenml = {{
             source = "zenml-io/zenml"
+            version = "{TERRAFORM_PROVIDER_VERSION_SPEC}"
         }}
     }}
 }}
@@ -302,7 +307,7 @@ provider "zenml" {{
 
 module "zenml_stack" {{
     source  = "zenml-io/zenml-stack/aws"
-
+    version = "{TERRAFORM_AWS_MODULE_VERSION_SPEC}"
     zenml_stack_name = "{self.stack_name}"
     zenml_stack_deployment = "{self.deployment_type}"
 }}

zenml/stack_deployments/azure_stack_deployment.py

@@ -18,6 +18,10 @@ from typing import ClassVar, Dict, List
 
 from zenml.enums import StackDeploymentProvider
 from zenml.models import StackDeploymentConfig
+from zenml.stack_deployments.constants import (
+    TERRAFORM_AZURE_MODULE_VERSION_SPEC,
+    TERRAFORM_PROVIDER_VERSION_SPEC,
+)
 from zenml.stack_deployments.stack_deployment import ZenMLCloudStackDeployment
 
 AZURE_DEPLOYMENT_TYPE = "azure-cloud-shell"
@@ -156,70 +160,70 @@ ZenML's access to your Azure subscription.
         """
         # Based on `az account list-locations -o table` on 16.07.2024
         return {
-            "(US) East US": "eastus",
-            "(US) South Central US": "southcentralus",
-            "(US) West US 2": "westus2",
-            "(US) West US 3": "westus3",
-            "(Asia Pacific) Australia East": "australiaeast",
-            "(Asia Pacific) Southeast Asia": "southeastasia",
-            "(Europe) North Europe": "northeurope",
-            "(Europe) Sweden Central": "swedencentral",
-            "(Europe) UK South": "uksouth",
-            "(Europe) West Europe": "westeurope",
-            "(US) Central US": "centralus",
             "(Africa) South Africa North": "southafricanorth",
+            "(Africa) South Africa West": "southafricawest",
+            "(Asia Pacific) Australia Central": "australiacentral",
+            "(Asia Pacific) Australia Central 2": "australiacentral2",
+            "(Asia Pacific) Australia East": "australiaeast",
+            "(Asia Pacific) Australia Southeast": "australiasoutheast",
             "(Asia Pacific) Central India": "centralindia",
             "(Asia Pacific) East Asia": "eastasia",
+            "(Asia Pacific) East Asia (Stage)": "eastasiastage",
             "(Asia Pacific) Japan East": "japaneast",
+            "(Asia Pacific) Japan West": "japanwest",
+            "(Asia Pacific) Jio India Central": "jioindiacentral",
+            "(Asia Pacific) Jio India West": "jioindiawest",
             "(Asia Pacific) Korea Central": "koreacentral",
+            "(Asia Pacific) Korea South": "koreasouth",
+            "(Asia Pacific) South India": "southindia",
+            "(Asia Pacific) Southeast Asia": "southeastasia",
+            "(Asia Pacific) Southeast Asia (Stage)": "southeastasiastage",
+            "(Asia Pacific) West India": "westindia",
             "(Canada) Canada Central": "canadacentral",
+            "(Canada) Canada East": "canadaeast",
             "(Europe) France Central": "francecentral",
+            "(Europe) France South": "francesouth",
+            "(Europe) Germany North": "germanynorth",
             "(Europe) Germany West Central": "germanywestcentral",
             "(Europe) Italy North": "italynorth",
+            "(Europe) North Europe": "northeurope",
             "(Europe) Norway East": "norwayeast",
+            "(Europe) Norway West": "norwaywest",
             "(Europe) Poland Central": "polandcentral",
             "(Europe) Spain Central": "spaincentral",
+            "(Europe) Sweden Central": "swedencentral",
             "(Europe) Switzerland North": "switzerlandnorth",
+            "(Europe) Switzerland West": "switzerlandwest",
+            "(Europe) UK South": "uksouth",
+            "(Europe) UK West": "ukwest",
+            "(Europe) West Europe": "westeurope",
             "(Mexico) Mexico Central": "mexicocentral",
-            "(Middle East) UAE North": "uaenorth",
-            "(South America) Brazil South": "brazilsouth",
             "(Middle East) Israel Central": "israelcentral",
             "(Middle East) Qatar Central": "qatarcentral",
+            "(Middle East) UAE Central": "uaecentral",
+            "(Middle East) UAE North": "uaenorth",
+            "(South America) Brazil South": "brazilsouth",
+            "(South America) Brazil Southeast": "brazilsoutheast",
+            "(South America) Brazil US": "brazilus",
+            "(US) Central US": "centralus",
             "(US) Central US (Stage)": "centralusstage",
+            "(US) Central US EUAP": "centraluseuap",
+            "(US) East US": "eastus",
             "(US) East US (Stage)": "eastusstage",
+            "(US) East US 2": "eastus2",
             "(US) East US 2 (Stage)": "eastus2stage",
+            "(US) East US 2 EUAP": "eastus2euap",
+            "(US) East US STG": "eastusstg",
+            "(US) North Central US": "northcentralus",
             "(US) North Central US (Stage)": "northcentralusstage",
+            "(US) South Central US": "southcentralus",
             "(US) South Central US (Stage)": "southcentralusstage",
+            "(US) West Central US": "westcentralus",
+            "(US) West US": "westus",
             "(US) West US (Stage)": "westusstage",
+            "(US) West US 2": "westus2",
             "(US) West US 2 (Stage)": "westus2stage",
-            "(Asia Pacific) East Asia (Stage)": "eastasiastage",
-            "(Asia Pacific) Southeast Asia (Stage)": "southeastasiastage",
-            "(South America) Brazil US": "brazilus",
-            "(US) East US 2": "eastus2",
-            "(US) East US STG": "eastusstg",
-            "(US) North Central US": "northcentralus",
-            "(US) West US": "westus",
-            "(Asia Pacific) Japan West": "japanwest",
-            "(Asia Pacific) Jio India West": "jioindiawest",
-            "(US) Central US EUAP": "centraluseuap",
-            "(US) East US 2 EUAP": "eastus2euap",
-            "(US) West Central US": "westcentralus",
-            "(Africa) South Africa West": "southafricawest",
-            "(Asia Pacific) Australia Central": "australiacentral",
-            "(Asia Pacific) Australia Central 2": "australiacentral2",
-            "(Asia Pacific) Australia Southeast": "australiasoutheast",
-            "(Asia Pacific) Jio India Central": "jioindiacentral",
-            "(Asia Pacific) Korea South": "koreasouth",
-            "(Asia Pacific) South India": "southindia",
-            "(Asia Pacific) West India": "westindia",
-            "(Canada) Canada East": "canadaeast",
-            "(Europe) France South": "francesouth",
-            "(Europe) Germany North": "germanynorth",
-            "(Europe) Norway West": "norwaywest",
-            "(Europe) Switzerland West": "switzerlandwest",
-            "(Europe) UK West": "ukwest",
-            "(Middle East) UAE Central": "uaecentral",
-            "(South America) Brazil Southeast": "brazilsoutheast",
+            "(US) West US 3": "westus3",
         }
 
     @classmethod
@@ -269,6 +273,7 @@ ZenML's access to your Azure subscription.
         }}
         zenml = {{
             source = "zenml-io/zenml"
+            version = "{TERRAFORM_PROVIDER_VERSION_SPEC}"
         }}
     }}
 }}
@@ -288,6 +293,7 @@ provider "zenml" {{
 
 module "zenml_stack" {{
     source  = "zenml-io/zenml-stack/azure"
+    version = "{TERRAFORM_AZURE_MODULE_VERSION_SPEC}"
 
     location = "{self.location or "eastus"}"
     zenml_stack_name = "{self.stack_name}"

zenml/stack_deployments/constants.py

@@ -0,0 +1,30 @@
+#  Copyright (c) ZenML GmbH 2025. All Rights Reserved.
+
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at:
+
+#       https://www.apache.org/licenses/LICENSE-2.0
+
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
+#  or implied. See the License for the specific language governing
+#  permissions and limitations under the License.
+"""Constants for the stack deployments."""
+
+# The ZenML Terraform provider compatible versions
+# (>= 2.0.0, < 3.0.0)
+TERRAFORM_PROVIDER_VERSION_SPEC = "~> 2.0"
+
+# The ZenML AWS Terraform module compatible versions
+# (>= 2.0.0, < 3.0.0)
+TERRAFORM_AWS_MODULE_VERSION_SPEC = "~> 2.0"
+
+# The ZenML GCP Terraform module compatible versions
+# (>= 2.0.0, < 3.0.0)
+TERRAFORM_GCP_MODULE_VERSION_SPEC = "~> 2.0"
+
+# The ZenML Azure Terraform module compatible versions
+# (>= 2.0.0, < 3.0.0)
+TERRAFORM_AZURE_MODULE_VERSION_SPEC = "~> 2.0"

zenml/stack_deployments/gcp_stack_deployment.py

@@ -18,6 +18,10 @@ from typing import ClassVar, Dict, List
 
 from zenml.enums import StackDeploymentProvider
 from zenml.models import StackDeploymentConfig
+from zenml.stack_deployments.constants import (
+    TERRAFORM_GCP_MODULE_VERSION_SPEC,
+    TERRAFORM_PROVIDER_VERSION_SPEC,
+)
 from zenml.stack_deployments.stack_deployment import (
     STACK_DEPLOYMENT_TERRAFORM,
     ZenMLCloudStackDeployment,
@@ -267,6 +271,7 @@ GCP project and to clean up the resources created by the stack by using
         }}
         zenml = {{
             source = "zenml-io/zenml"
+            version = "{TERRAFORM_PROVIDER_VERSION_SPEC}"
         }}
     }}
 }}
@@ -283,6 +288,7 @@ provider "zenml" {{
 
 module "zenml_stack" {{
     source  = "zenml-io/zenml-stack/gcp"
+    version = "{TERRAFORM_GCP_MODULE_VERSION_SPEC}"
 
     zenml_stack_name = "{self.stack_name}"
     zenml_stack_deployment = "{self.deployment_type}"

zenml/steps/entrypoint_function_utils.py

@@ -174,7 +174,7 @@ class EntrypointFunctionDefinition(NamedTuple):
                 f"'{key}' is not JSON serializable and can not be passed as "
                 "a parameter. This input can either be provided by the "
                 "output of another step or as an external artifact: "
-                "https://docs.zenml.io/user-guide/starter-guide/manage-artifacts#managing-artifacts-not-produced-by-zenml-pipelines"
+                "https://docs.zenml.io/user-guides/starter-guide/manage-artifacts#managing-artifacts-not-produced-by-zenml-pipelines"
             )
 
         try: