zenml-nightly 0.80.0.dev20250321__tar.gz → 0.80.0.dev20250322__tar.gz

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.80.0.dev20250321
+Version: 0.80.0.dev20250322
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -54,7 +54,7 @@ Requires-Dist: azure-storage-blob (>=12.0.0) ; extra == "connectors-azure"
 Requires-Dist: bandit (>=1.7.5,<2.0.0) ; extra == "dev"
 Requires-Dist: bcrypt (==4.0.1)
 Requires-Dist: boto3 (>=1.16.0) ; extra == "secrets-aws" or extra == "connectors-aws"
-Requires-Dist: click (>=8.0.1,<8.1.4)
+Requires-Dist: click (>=8.0.1,<8.1.8)
 Requires-Dist: cloudpickle (>=2.0.0,<3)
 Requires-Dist: copier (>=8.1.0) ; extra == "templates"
 Requires-Dist: coverage[toml] (>=5.5,<6.0) ; extra == "dev"

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zenml-nightly"
-version = "0.80.0.dev20250321"
+version = "0.80.0.dev20250322"
 packages = [{ include = "zenml", from = "src" }]
 description = "ZenML: Write production-ready ML code."
 authors = ["ZenML GmbH <info@zenml.io>"]
@@ -43,7 +43,7 @@ zenml = "zenml.cli.cli:cli"
 [tool.poetry.dependencies]
 alembic = { version = "~1.8.1" }
 bcrypt = { version = "4.0.1" }
-click = "^8.0.1,<8.1.4"
+click = "^8.0.1,<8.1.8"
 cloudpickle = ">=2.0.0,<3"
 distro = "^1.6.0"
 docker = "~7.1.0"

zenml_nightly-0.80.0.dev20250322/src/zenml/VERSION

@@ -0,0 +1 @@
+0.80.0.dev20250322

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/cli/project.py

@@ -38,7 +38,7 @@ def project() -> None:
 @project.command("list")
 @list_options(ProjectFilter)
 @click.pass_context
-def list_projects(ctx: click.Context, **kwargs: Any) -> None:
+def list_projects(ctx: click.Context, /, **kwargs: Any) -> None:
     """List all projects.
 
     Args:

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/cli/service_accounts.py

@@ -187,7 +187,7 @@ def describe_service_account(service_account_name_or_id: str) -> None:
 @service_account.command("list")
 @list_options(ServiceAccountFilter)
 @click.pass_context
-def list_service_accounts(ctx: click.Context, **kwargs: Any) -> None:
+def list_service_accounts(ctx: click.Context, /, **kwargs: Any) -> None:
     """List all users.
 
     Args:
@@ -384,7 +384,7 @@ def describe_api_key(service_account_name_or_id: str, name_or_id: str) -> None:
 @api_key.command("list", help="List all API keys.")
 @list_options(APIKeyFilter)
 @click.pass_obj
-def list_api_keys(service_account_name_or_id: str, **kwargs: Any) -> None:
+def list_api_keys(service_account_name_or_id: str, /, **kwargs: Any) -> None:
     """List all API keys.
 
     Args:

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/cli/service_connectors.py

@@ -982,7 +982,7 @@ def register_service_connector(
 )
 @click.pass_context
 def list_service_connectors(
-    ctx: click.Context, labels: Optional[List[str]] = None, **kwargs: Any
+    ctx: click.Context, /, labels: Optional[List[str]] = None, **kwargs: Any
 ) -> None:
     """List all service connectors.
 

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/cli/stack.py

@@ -937,7 +937,7 @@ def rename_stack(
 @stack.command("list")
 @list_options(StackFilter)
 @click.pass_context
-def list_stacks(ctx: click.Context, **kwargs: Any) -> None:
+def list_stacks(ctx: click.Context, /, **kwargs: Any) -> None:
     """List all stacks that fulfill the filter requirements.
 
     Args:

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/cli/stack_components.py

@@ -155,7 +155,7 @@ def generate_stack_component_list_command(
     @list_options(ComponentFilter)
     @click.pass_context
     def list_stack_components_command(
-        ctx: click.Context, **kwargs: Any
+        ctx: click.Context, /, **kwargs: Any
     ) -> None:
         """Prints a table of stack components.
 

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/cli/user_management.py

@@ -79,7 +79,7 @@ def describe_user(user_name_or_id: Optional[str] = None) -> None:
 @user.command("list")
 @list_options(UserFilter)
 @click.pass_context
-def list_users(ctx: click.Context, **kwargs: Any) -> None:
+def list_users(ctx: click.Context, /, **kwargs: Any) -> None:
     """List all users.
 
     Args:

zenml_nightly-0.80.0.dev20250322/src/zenml/integrations/__init__.py

@@ -0,0 +1,19 @@
+#  Copyright (c) ZenML GmbH 2022. 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.
+"""ZenML integrations module.
+
+The ZenML integrations module contains sub-modules for each integration that we
+support. This includes orchestrators like Apache Airflow, visualization tools
+like the ``facets`` library, as well as deep learning libraries like PyTorch.
+"""

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py

@@ -223,45 +223,52 @@ class KubernetesOrchestrator(ContainerizedOrchestrator):
             msg = f"'{self.name}' Kubernetes orchestrator error: "
 
             if not self.connector:
-                if not kubernetes_context:
+                if kubernetes_context:
+                    contexts, active_context = self.get_kubernetes_contexts()
+
+                    if kubernetes_context not in contexts:
+                        return False, (
+                            f"{msg}could not find a Kubernetes context named "
+                            f"'{kubernetes_context}' in the local "
+                            "Kubernetes configuration. Please make sure that "
+                            "the Kubernetes cluster is running and that the "
+                            "kubeconfig file is configured correctly. To list "
+                            "all configured contexts, run:\n\n"
+                            "  `kubectl config get-contexts`\n"
+                        )
+                    if kubernetes_context != active_context:
+                        logger.warning(
+                            f"{msg}the Kubernetes context "
+                            f"'{kubernetes_context}' configured for the "
+                            f"Kubernetes orchestrator is not the same as the "
+                            f"active context in the local Kubernetes "
+                            f"configuration. If this is not deliberate, you "
+                            f"should update the orchestrator's "
+                            f"`kubernetes_context` field by running:\n\n"
+                            f"  `zenml orchestrator update {self.name} "
+                            f"--kubernetes_context={active_context}`\n"
+                            f"To list all configured contexts, run:\n\n"
+                            f"  `kubectl config get-contexts`\n"
+                            f"To set the active context to be the same as the "
+                            f"one configured in the Kubernetes orchestrator "
+                            f"and silence this warning, run:\n\n"
+                            f"  `kubectl config use-context "
+                            f"{kubernetes_context}`\n"
+                        )
+                elif self.config.incluster:
+                    # No service connector or kubernetes_context is needed when
+                    # the orchestrator is being used from within a Kubernetes
+                    # cluster.
+                    pass
+                else:
                     return False, (
-                        f"{msg}you must either link this stack component to a "
+                        f"{msg}you must either link this orchestrator to a "
                         "Kubernetes service connector (see the 'zenml "
-                        "orchestrator connect' CLI command) or explicitly set "
+                        "orchestrator connect' CLI command), explicitly set "
                         "the `kubernetes_context` attribute to the name of the "
                         "Kubernetes config context pointing to the cluster "
-                        "where you would like to run pipelines."
-                    )
-
-                contexts, active_context = self.get_kubernetes_contexts()
-
-                if kubernetes_context not in contexts:
-                    return False, (
-                        f"{msg}could not find a Kubernetes context named "
-                        f"'{kubernetes_context}' in the local "
-                        "Kubernetes configuration. Please make sure that the "
-                        "Kubernetes cluster is running and that the kubeconfig "
-                        "file is configured correctly. To list all configured "
-                        "contexts, run:\n\n"
-                        "  `kubectl config get-contexts`\n"
-                    )
-                if kubernetes_context != active_context:
-                    logger.warning(
-                        f"{msg}the Kubernetes context '{kubernetes_context}' "  # nosec
-                        f"configured for the Kubernetes orchestrator is not "
-                        f"the same as the active context in the local "
-                        f"Kubernetes configuration. If this is not deliberate,"
-                        f" you should update the orchestrator's "
-                        f"`kubernetes_context` field by running:\n\n"
-                        f"  `zenml orchestrator update {self.name} "
-                        f"--kubernetes_context={active_context}`\n"
-                        f"To list all configured contexts, run:\n\n"
-                        f"  `kubectl config get-contexts`\n"
-                        f"To set the active context to be the same as the one "
-                        f"configured in the Kubernetes orchestrator and "
-                        f"silence this warning, run:\n\n"
-                        f"  `kubectl config use-context "
-                        f"{kubernetes_context}`\n"
+                        "where you would like to run pipelines, or set the "
+                        "`incluster` attribute to `True`."
                     )
 
             silence_local_validations_msg = (

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/integrations/registry.py

@@ -13,6 +13,8 @@
 #  permissions and limitations under the License.
 """Implementation of a registry to track ZenML integrations."""
 
+import importlib
+import os
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Type
 
 from zenml.exceptions import IntegrationError
@@ -30,6 +32,7 @@ class IntegrationRegistry(object):
     def __init__(self) -> None:
         """Initializing the integration registry."""
         self._integrations: Dict[str, Type["Integration"]] = {}
+        self._initialized = False
 
     @property
     def integrations(self) -> Dict[str, Type["Integration"]]:
@@ -38,6 +41,7 @@ class IntegrationRegistry(object):
         Returns:
             A dict of integration key to type of `Integration`.
         """
+        self._initialize()
         return self._integrations
 
     @integrations.setter
@@ -68,8 +72,33 @@ class IntegrationRegistry(object):
         """
         self._integrations[key] = type_
 
+    def _initialize(self) -> None:
+        """Method to register all integrations."""
+        if self._initialized:
+            return
+        self._initialized = True
+
+        # Load all submodules in the integrations module
+        integrations_dir = os.path.dirname(__file__)
+        for file in os.listdir(integrations_dir):
+            full_path = os.path.join(integrations_dir, file)
+            # Skip anything that isn't a directory
+            if not os.path.isdir(full_path):
+                continue
+            # Skip anything that doesn't have a __init__.py file
+            if not os.path.exists(os.path.join(full_path, "__init__.py")):
+                continue
+            # Import the module
+            module_path = f"zenml.integrations.{file}"
+            try:
+                importlib.import_module(module_path)
+            except ImportError:
+                logger.exception(f"Failed to import module `{module_path}`.")
+                continue
+
     def activate_integrations(self) -> None:
         """Method to activate the integrations with are registered in the registry."""
+        self._initialize()
         for name, integration in self._integrations.items():
             if integration.check_installation():
                 logger.debug(f"Activating integration `{name}`...")
@@ -85,6 +114,7 @@ class IntegrationRegistry(object):
         Returns:
             A list of all possible integrations.
         """
+        self._initialize()
         return [name for name in self._integrations]
 
     def select_integration_requirements(
@@ -104,6 +134,7 @@ class IntegrationRegistry(object):
         Raises:
             KeyError: If the integration is not found.
         """
+        self._initialize()
         if integration_name:
             if integration_name in self.list_integration_names:
                 return self._integrations[integration_name].get_requirements(
@@ -141,6 +172,7 @@ class IntegrationRegistry(object):
         Raises:
             KeyError: If the integration is not found.
         """
+        self._initialize()
         if integration_name:
             if integration_name in self.list_integration_names:
                 return self._integrations[
@@ -173,6 +205,7 @@ class IntegrationRegistry(object):
         Raises:
             KeyError: If the integration is not found.
         """
+        self._initialize()
         if integration_name in self.list_integration_names:
             return self._integrations[integration_name].check_installation()
         elif not integration_name:
@@ -194,6 +227,7 @@ class IntegrationRegistry(object):
         Returns:
             List of installed integrations.
         """
+        self._initialize()
         return [
             name
             for name, integration in integration_registry.integrations.items()

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/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_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/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_nightly-0.80.0.dev20250322/src/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_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/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_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/zen_server/routers/stack_deployment_endpoints.py

@@ -40,6 +40,7 @@ from zenml.zen_server.rbac.models import Action, ResourceType
 from zenml.zen_server.rbac.utils import verify_permission
 from zenml.zen_server.utils import (
     handle_exceptions,
+    server_config,
 )
 
 router = APIRouter(
@@ -105,10 +106,17 @@ def get_stack_deployment_config(
     verify_permission(resource_type=ResourceType.STACK, action=Action.CREATE)
 
     stack_deployment_class = get_stack_deployment_class(provider)
-    # Get the base server URL used to call this FastAPI endpoint
-    url = request.url.replace(path="").replace(query="")
-    # Use HTTPS for the URL
-    url = url.replace(scheme="https")
+
+    config = server_config()
+    if config.server_url:
+        url = config.server_url
+    else:
+        # Get the base server URL used to call this FastAPI endpoint
+        url = str(
+            request.url.replace(path="")
+            .replace(query="")
+            .replace(scheme="https")
+        )
 
     token = auth_context.access_token
     assert token is not None

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/zen_stores/migrations/versions/288f4fb6e112_make_tags_user_scoped.py

@@ -1,7 +1,7 @@
 """make tags user scoped [288f4fb6e112].
 
 Revision ID: 288f4fb6e112
-Revises: 3b1776345020
+Revises: 41b28cae31ce
 Create Date: 2025-02-19 15:16:42.954792
 
 """
@@ -13,7 +13,7 @@ from sqlalchemy.orm import Session
 
 # revision identifiers, used by Alembic.
 revision = "288f4fb6e112"
-down_revision = "3b1776345020"
+down_revision = "41b28cae31ce"
 branch_labels = None
 depends_on = None
 

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/zen_stores/migrations/versions/41b28cae31ce_make_artifacts_workspace_scoped.py

@@ -1,7 +1,7 @@
 """make artifacts workspace scoped [41b28cae31ce].
 
 Revision ID: 41b28cae31ce
-Revises: 288f4fb6e112
+Revises: 3b1776345020
 Create Date: 2025-02-19 23:23:08.133826
 
 """
@@ -21,7 +21,7 @@ from zenml.constants import (
 
 # revision identifiers, used by Alembic.
 revision = "41b28cae31ce"
-down_revision = "288f4fb6e112"
+down_revision = "3b1776345020"
 branch_labels = None
 depends_on = None
 

{zenml_nightly-0.80.0.dev20250321 → zenml_nightly-0.80.0.dev20250322}/src/zenml/zen_stores/migrations/versions/f1d723fd723b_add_secret_private_attr.py

@@ -1,7 +1,7 @@
 """add secret private attr [f1d723fd723b].
 
 Revision ID: f1d723fd723b
-Revises: 41b28cae31ce
+Revises: 288f4fb6e112
 Create Date: 2025-02-21 19:40:14.596681
 
 """
@@ -11,7 +11,7 @@ from alembic import op
 
 # revision identifiers, used by Alembic.
 revision = "f1d723fd723b"
-down_revision = "41b28cae31ce"
+down_revision = "288f4fb6e112"
 branch_labels = None
 depends_on = None
 

zenml_nightly-0.80.0.dev20250321/src/zenml/VERSION

@@ -1 +0,0 @@
-0.80.0.dev20250321