zenml-nightly 0.81.0.dev20250424__py3-none-any.whl → 0.81.0.dev20250426__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.81.0.dev20250424
+0.81.0.dev20250426

zenml/config/compiler.py

@@ -395,6 +395,25 @@ class Compiler:
                     f"The settings class {settings_instance.__class__} can not "
                     f"be specified on a {configuration_level.name} level."
                 )
+
+            if settings_instance.model_extra:
+                logger.warning(
+                    "Ignoring invalid setting attributes `%s` defined for key `%s`.",
+                    list(settings_instance.model_extra),
+                    key,
+                )
+                settings_instance = settings_instance.model_validate(
+                    settings_instance.model_dump(
+                        exclude=set(settings_instance.model_extra),
+                        exclude_unset=True,
+                    )
+                )
+
+            if not settings_instance.model_fields_set:
+                # There are no values defined on the settings instance, don't
+                # include them in the deployment
+                continue
+
             validated_settings[key] = settings_instance
 
         return validated_settings

zenml/constants.py

@@ -200,6 +200,7 @@ ENV_ZENML_RUN_SINGLE_STEPS_WITHOUT_STACK = (
 ENV_ZENML_PREVENT_CLIENT_SIDE_CACHING = "ZENML_PREVENT_CLIENT_SIDE_CACHING"
 ENV_ZENML_DISABLE_CREDENTIALS_DISK_CACHING = "DISABLE_CREDENTIALS_DISK_CACHING"
 ENV_ZENML_RUNNER_IMAGE_DISABLE_UV = "ZENML_RUNNER_IMAGE_DISABLE_UV"
+ENV_ZENML_RUNNER_POD_TIMEOUT = "ZENML_RUNNER_POD_TIMEOUT"
 ENV_ZENML_WORKLOAD_TOKEN_EXPIRATION_LEEWAY = (
     "ZENML_WORKLOAD_TOKEN_EXPIRATION_LEEWAY"
 )

zenml/zen_server/template_execution/utils.py

@@ -21,7 +21,9 @@ from zenml.constants import (
     ENV_ZENML_ACTIVE_PROJECT_ID,
     ENV_ZENML_ACTIVE_STACK_ID,
     ENV_ZENML_RUNNER_IMAGE_DISABLE_UV,
+    ENV_ZENML_RUNNER_POD_TIMEOUT,
     handle_bool_env_var,
+    handle_int_env_var,
 )
 from zenml.enums import ExecutionStatus, StackComponentType, StoreType
 from zenml.logger import get_logger
@@ -192,6 +194,10 @@ def run_template(
             message="Starting pipeline run.",
         )
 
+        runner_timeout = handle_int_env_var(
+            ENV_ZENML_RUNNER_POD_TIMEOUT, default=60
+        )
+
         # could do this same thing with a step operator, but we need some
         # minor changes to the abstract interface to support that.
         workload_manager().run(
@@ -200,7 +206,7 @@ def run_template(
             command=command,
             arguments=args,
             environment=environment,
-            timeout_in_seconds=30,
+            timeout_in_seconds=runner_timeout,
             sync=True,
         )
         workload_manager().log(

zenml/zen_stores/base_zen_store.py

@@ -450,20 +450,6 @@ class BaseZenStore(
         """
         return os.getenv(ENV_ZENML_DEFAULT_PROJECT_NAME, DEFAULT_PROJECT_NAME)
 
-    def _get_default_project(self) -> ProjectResponse:
-        """Get the default project.
-
-        Raises:
-            KeyError: If the default project doesn't exist.
-
-        Returns:
-            The default project.
-        """
-        try:
-            return self.get_project(self._default_project_name)
-        except KeyError:
-            raise KeyError("Unable to find default project.")
-
     def _get_default_stack(
         self,
     ) -> StackResponse:

zenml/zen_stores/sql_zen_store.py

@@ -1155,19 +1155,7 @@ class SqlZenStore(BaseZenStore):
 
     def _initialize_database(self) -> None:
         """Initialize the database if not already initialized."""
-        # When running in a Pro ZenML server, the default project is not
-        # created on database initialization but on server onboarding.
-        create_default_project = True
-        if ENV_ZENML_SERVER in os.environ:
-            from zenml.config.server_config import ServerConfiguration
-
-            if (
-                ServerConfiguration.get_server_config().deployment_type
-                == ServerDeploymentType.CLOUD
-            ):
-                create_default_project = False
-
-        if create_default_project:
+        if self._default_project_enabled:
             # Make sure the default project exists
             self._get_or_create_default_project()
         # Make sure the default stack exists
@@ -9440,7 +9428,8 @@ class SqlZenStore(BaseZenStore):
                 session=session,
             )
             if (
-                existing_project.name == self._default_project_name
+                self._default_project_enabled
+                and existing_project.name == self._default_project_name
                 and "name" in project_update.model_fields_set
                 and project_update.name != existing_project.name
             ):
@@ -9481,7 +9470,10 @@ class SqlZenStore(BaseZenStore):
                 schema_class=ProjectSchema,
                 session=session,
             )
-            if project.name == self._default_project_name:
+            if (
+                self._default_project_enabled
+                and project.name == self._default_project_name
+            ):
                 raise IllegalOperationError(
                     "The default project cannot be deleted."
                 )
@@ -9542,6 +9534,27 @@ class SqlZenStore(BaseZenStore):
                 ProjectRequest(name=default_project_name)
             )
 
+    @property
+    def _default_project_enabled(self) -> bool:
+        """Check if the default project is enabled.
+
+        When running in a Pro ZenML server, the default project is not enabled.
+
+        Returns:
+            True if the default project is enabled, False otherwise.
+        """
+        default_project_enabled = True
+        if ENV_ZENML_SERVER in os.environ:
+            from zenml.config.server_config import ServerConfiguration
+
+            if (
+                ServerConfiguration.get_server_config().deployment_type
+                == ServerDeploymentType.CLOUD
+            ):
+                default_project_enabled = False
+
+        return default_project_enabled
+
     # =======================
     # Internal helper methods
     # =======================
@@ -9915,7 +9928,7 @@ class SqlZenStore(BaseZenStore):
                 session=session,
             )
             project_id = project.id
-        else:
+        elif self._default_project_enabled:
             # Use the default project as a last resort.
             try:
                 project = self._get_schema_by_name_or_id(
@@ -9926,6 +9939,8 @@ class SqlZenStore(BaseZenStore):
                 project_id = project.id
             except KeyError:
                 raise ValueError("Project scope missing from the filter")
+        else:
+            raise ValueError("Project scope missing from the filter")
 
         filter_model.project = project_id
 

{zenml_nightly-0.81.0.dev20250424.dist-info → zenml_nightly-0.81.0.dev20250426.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.81.0.dev20250424
+Version: 0.81.0.dev20250426
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -283,7 +283,7 @@ def rag_pipeline() -> str:
     index = index_generator(embeddings)
     return index
 ```
-![Running a ZenML pipeline](/docs/book/.gitbook/assets/readme_simple_pipeline.gif)
+![Running a ZenML pipeline](docs/book/.gitbook/assets/readme_simple_pipeline.gif)
 
 ### Easily provision an MLOps stack or reuse your existing infrastructure
 
@@ -291,7 +291,7 @@ The framework is a gentle entry point for practitioners to build complex ML pipe
 
 ZenML provides different features to aid people to get started quickly on a remote setting as well. If you want to deploy a remote stack from scratch on your selected cloud provider, you can use the 1-click deployment feature either through the dashboard:
 
-![Running a ZenML pipeline](/docs/book/.gitbook/assets/one-click-deployment.gif)
+![Running a ZenML pipeline](docs/book/.gitbook/assets/one-click-deployment.gif)
 
 Or, through our CLI command:
 
@@ -329,7 +329,7 @@ def training(...):
 	...
 ```
 
-![Workloads with ZenML](/docs/book/.gitbook/assets/readme_compute.gif)
+![Workloads with ZenML](docs/book/.gitbook/assets/readme_compute.gif)
 
 ### Track models, pipeline, and artifacts
 
@@ -346,7 +346,7 @@ def deploy_rag(index_id: str) -> str:
     return deployment_id
 ```
 
-![Exploring ZenML Models](/docs/book/.gitbook/assets/readme_mcp.gif)
+![Exploring ZenML Models](docs/book/.gitbook/assets/readme_mcp.gif)
 
 ## 🚀 Key LLMOps Capabilities
 
@@ -354,7 +354,7 @@ def deploy_rag(index_id: str) -> str:
 **Build production-ready retrieval systems**  
 
 <div align="center">
-  <img src="/docs/book/.gitbook/assets/rag_zenml_home.png" width="800" alt="RAG Pipeline">
+  <img src="docs/book/.gitbook/assets/rag_zenml_home.png" width="800" alt="RAG Pipeline">
 </div>
 
 ZenML tracks document ingestion, embedding versions, and query patterns. Implement feedback loops and:
@@ -367,7 +367,7 @@ ZenML tracks document ingestion, embedding versions, and query patterns. Impleme
 **Confidence in model updates**
 
 <div align="center">
-  <img src="/docs/book/.gitbook/assets/finetune_zenml_home.png" width="800" alt="Finetuning Pipeline">
+  <img src="docs/book/.gitbook/assets/finetune_zenml_home.png" width="800" alt="Finetuning Pipeline">
 </div>
 
 Maintain full lineage of SLM/LLM training runs:
@@ -390,7 +390,7 @@ def train_and_deploy(training_df: pd.DataFrame) -> bento.Bento
 	return bento
 ```
 
-![Exploring ZenML Integrations](/docs/book/.gitbook/assets/readme_integrations.gif)
+![Exploring ZenML Integrations](docs/book/.gitbook/assets/readme_integrations.gif)
 
 ## 🔄 Your LLM Framework Isn't Enough for Production
 
@@ -447,7 +447,7 @@ ZenML has a [VS Code extension](https://marketplace.visualstudio.com/items?itemN
 <details>
   <summary>🖥️ VS Code Extension in Action!</summary>
   <div align="center">
-  <img width="60%" src="/docs/book/.gitbook/assets/zenml-extension-shortened.gif" alt="ZenML Extension">
+  <img width="60%" src="docs/book/.gitbook/assets/zenml-extension-shortened.gif" alt="ZenML Extension">
 </div>
 </details>
 

{zenml_nightly-0.81.0.dev20250424.dist-info → zenml_nightly-0.81.0.dev20250426.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=xC_BFuquXquHB86Esw9B286z8kOV2CSSQJPW27TlBvg,19
+zenml/VERSION,sha256=ZOGvzbsgxYXCW1IR-ANCD5Xgf5KUHEGMNbYPf32lFlo,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
@@ -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=Jsng7ebpeaRbzXlbszU-uYkaVihgQ4OrD839yWwD3ZY,5126
-zenml/config/compiler.py,sha256=y7WEGOOuGvblW7nUD8GzLmBK9Wb7NL2iy7ehbdR8TMs,23384
+zenml/config/compiler.py,sha256=jzyyTmM8nTxtpIVTSjZ-jrVOBJ1roVoJ3fs1A1nwv9M,24130
 zenml/config/constants.py,sha256=QvSgMwXWxtspcJ45CrFDP1ZY3w6gS3bIhXLOtIDAbZA,713
 zenml/config/docker_settings.py,sha256=KTUR9FaRn3ivcagzFgpLe4eev8fvd5Wq1oV8wGNMB-g,13322
 zenml/config/global_config.py,sha256=ZD3WodfcWMBJZOl1FNn3ztzwilGDDv9EKdHClLqSO8s,29562
@@ -85,7 +85,7 @@ zenml/config/step_run_info.py,sha256=KiVRSTtKmZ1GbvseDTap2imr7XwMHD3jSFVpyLNEK1I
 zenml/config/store_config.py,sha256=Cla5p5dTB6nNlo8_OZDs9hod5hspi64vxwtZj882XgU,3559
 zenml/config/strict_base_model.py,sha256=iHnO9qOmLUP_eiy9IjRr3JjIs1l1I_CsRQ76EyAneYU,860
 zenml/console.py,sha256=hj_KerPQKwnyKACj0ehSqUQX0mGVCJBKE1QvCt6ik3A,1160
-zenml/constants.py,sha256=DbOD7MmcwOCi1cq-jiF8PhWqHUZanHaQg1jj9W8Jmvk,16373
+zenml/constants.py,sha256=ZVwWVxE2h54cNbJDyujJmhAc8w3J7I9MkWHOfP4HN9A,16431
 zenml/container_registries/__init__.py,sha256=ZSPbBIOnzhg88kQSpYgKe_POLuru14m629665-kAVAA,2200
 zenml/container_registries/azure_container_registry.py,sha256=t1sfDa94Vzbyqtb1iPFNutJ2EXV5_p9CUNITasoiQ70,2667
 zenml/container_registries/base_container_registry.py,sha256=6c2e32wuqxYHJXm5OV2LY1MtX9yopB7WZtes9fmTAz0,7625
@@ -1063,12 +1063,12 @@ zenml/zen_server/routers/webhook_endpoints.py,sha256=KOJsuykv_TMjL3oEItpC4OWWP75
 zenml/zen_server/secure_headers.py,sha256=glh6QujnjyeoH1_FK-tAS-105G-qKS_34AqSzqJ6TRc,4182
 zenml/zen_server/template_execution/__init__.py,sha256=79knXLKfegsvVSVSWecpqrepq6iAavTUA4hKuiDk-WE,613
 zenml/zen_server/template_execution/runner_entrypoint_configuration.py,sha256=Y8aYJhqqs8Kv8I1q-dM1WemS5VBIfyoaaYH_YkzC7iY,1541
-zenml/zen_server/template_execution/utils.py,sha256=nDf7CTY5LQPHczBP3m4TgQAJub4qP-VwVBO-vlNFD18,17354
+zenml/zen_server/template_execution/utils.py,sha256=UGMFeyVB8Os2awUfXsvKvDLdMiT0uxRj7ErF-8RyhA0,17533
 zenml/zen_server/template_execution/workload_manager_interface.py,sha256=CL9c7z8ajuZE01DaHmdCDCZmsroDcTarvN-nE8jv6qQ,2590
 zenml/zen_server/utils.py,sha256=Jc2Q4UBaYG2ruHdsN9JmbOWfWU_eWD9wTBBEgcGAbqg,17439
 zenml/zen_server/zen_server_api.py,sha256=ALyv5096frXXRNySegEtsmkDbHLLqHd402bbQI7RnII,17941
 zenml/zen_stores/__init__.py,sha256=6LTgH6XwDeDxKqVJ1JTfGhmS8II1NLopPloINGmdyI0,691
-zenml/zen_stores/base_zen_store.py,sha256=kazsQSYD7fl51rBxXPpaXRsrMFOpOo5zCF5nN5qoMkc,16520
+zenml/zen_stores/base_zen_store.py,sha256=AplsW2NR-G9_CU54XvNTQJo4W0KJ5TJV22cjKW4n2BY,16124
 zenml/zen_stores/migrations/README.md,sha256=x04jsb6EOP6PBEGMQlDELiqKEham2O-iztAs9AylMFc,4898
 zenml/zen_stores/migrations/__init__.py,sha256=N9CHfdz0AZ6KniQ450VCIV3H0CuWtx83AloYy82woho,657
 zenml/zen_stores/migrations/alembic.py,sha256=JDqx7Md6DxnHtP3xrZG1I0cNv6NyTR0oO3tPRUPaS2I,7455
@@ -1313,11 +1313,11 @@ zenml/zen_stores/secrets_stores/hashicorp_secrets_store.py,sha256=NfW1EHIA99lseb
 zenml/zen_stores/secrets_stores/secrets_store_interface.py,sha256=Q2Jbnt2Pp7NGlR-u1YBfRZV2g8su2Fd0ArBMdksAE-Q,2819
 zenml/zen_stores/secrets_stores/service_connector_secrets_store.py,sha256=S87ne23D08PAwtfRVlVnBn8R0ilTpEh6r8blauNV5WQ,6941
 zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=nEO0bAPlULBLxLVk-UTRIZiUeVpATggo8qCsKmgEU1E,8788
-zenml/zen_stores/sql_zen_store.py,sha256=7ISpj0CC31xb66ehW1PZZ2Yapk3Sxu-CiuOGZlLqGQA,441658
+zenml/zen_stores/sql_zen_store.py,sha256=diS6mZwE0A5v2wKf69CDtwCr1Bveot8QSQc-WE44rw4,442103
 zenml/zen_stores/template_utils.py,sha256=GWBP5QEOyvhzndS_MLPmvh28sQaOPpPoZFXCIX9CRL4,9065
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.81.0.dev20250424.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.81.0.dev20250424.dist-info/METADATA,sha256=JhanTnu1Knd_WiKokvZmb9gEhoSDyiHwP5hl_enzx7w,24312
-zenml_nightly-0.81.0.dev20250424.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-zenml_nightly-0.81.0.dev20250424.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.81.0.dev20250424.dist-info/RECORD,,
+zenml_nightly-0.81.0.dev20250426.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.81.0.dev20250426.dist-info/METADATA,sha256=Hh6sfwT9wRec5s3iNU0sEPF0dN8AZS-vdtbKLouPnn8,24304
+zenml_nightly-0.81.0.dev20250426.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+zenml_nightly-0.81.0.dev20250426.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.81.0.dev20250426.dist-info/RECORD,,