zenml-nightly 0.81.0.dev20250424__tar.gz → 0.81.0.dev20250426__tar.gz

{zenml_nightly-0.81.0.dev20250424 → zenml_nightly-0.81.0.dev20250426}/PKG-INFO

@@ -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 → zenml_nightly-0.81.0.dev20250426}/README.md

@@ -137,7 +137,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
 
@@ -145,7 +145,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:
 
@@ -183,7 +183,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
 
@@ -200,7 +200,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
 
@@ -208,7 +208,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:
@@ -221,7 +221,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:
@@ -244,7 +244,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
 
@@ -301,7 +301,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 → zenml_nightly-0.81.0.dev20250426}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zenml-nightly"
-version = "0.81.0.dev20250424"
+version = "0.81.0.dev20250426"
 packages = [{ include = "zenml", from = "src" }]
 description = "ZenML: Write production-ready ML code."
 authors = ["ZenML GmbH <info@zenml.io>"]

zenml_nightly-0.81.0.dev20250426/src/zenml/VERSION

@@ -0,0 +1 @@
+0.81.0.dev20250426

{zenml_nightly-0.81.0.dev20250424 → zenml_nightly-0.81.0.dev20250426}/src/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_nightly-0.81.0.dev20250424 → zenml_nightly-0.81.0.dev20250426}/src/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_nightly-0.81.0.dev20250424 → zenml_nightly-0.81.0.dev20250426}/src/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_nightly-0.81.0.dev20250424 → zenml_nightly-0.81.0.dev20250426}/src/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_nightly-0.81.0.dev20250424 → zenml_nightly-0.81.0.dev20250426}/src/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/src/zenml/VERSION

@@ -1 +0,0 @@
-0.81.0.dev20250424