zenml-nightly 0.80.1.dev20250407__py3-none-any.whl → 0.80.1.dev20250409__py3-none-any.whl

zenml/zen_server/template_execution/utils.py

@@ -20,6 +20,8 @@ from zenml.config.step_configurations import Step, StepConfiguration
 from zenml.constants import (
     ENV_ZENML_ACTIVE_PROJECT_ID,
     ENV_ZENML_ACTIVE_STACK_ID,
+    ENV_ZENML_RUNNER_IMAGE_DISABLE_UV,
+    handle_bool_env_var,
 )
 from zenml.enums import ExecutionStatus, StackComponentType, StoreType
 from zenml.logger import get_logger
@@ -148,35 +150,36 @@ def run_template(
         deployment_id=new_deployment.id
     )
 
-    def _task() -> None:
-        if build.python_version:
-            version_info = version.parse(build.python_version)
-            python_version = f"{version_info.major}.{version_info.minor}"
-        else:
-            python_version = (
-                f"{sys.version_info.major}.{sys.version_info.minor}"
-            )
+    if build.python_version:
+        version_info = version.parse(build.python_version)
+        python_version = f"{version_info.major}.{version_info.minor}"
+    else:
+        python_version = f"{sys.version_info.major}.{sys.version_info.minor}"
 
-        (
-            pypi_requirements,
-            apt_packages,
-        ) = requirements_utils.get_requirements_for_stack(
-            stack=stack, python_version=python_version
-        )
+    (
+        pypi_requirements,
+        apt_packages,
+    ) = requirements_utils.get_requirements_for_stack(
+        stack=stack, python_version=python_version
+    )
 
-        dockerfile = generate_dockerfile(
-            pypi_requirements=pypi_requirements,
-            apt_packages=apt_packages,
-            zenml_version=zenml_version,
-            python_version=python_version,
-        )
+    dockerfile = generate_dockerfile(
+        pypi_requirements=pypi_requirements,
+        apt_packages=apt_packages,
+        zenml_version=zenml_version,
+        python_version=python_version,
+    )
 
-        # building a docker image with requirements and apt packages from the
-        # stack only (no code). Ideally, only orchestrator requirements should
-        # be added to the docker image, but we have to instantiate the entire
-        # stack to get the orchestrator to run pipelines.
-        image_hash = generate_image_hash(dockerfile=dockerfile)
+    # Building a docker image with requirements and apt packages from the
+    # stack only (no code). Ideally, only orchestrator requirements should
+    # be added to the docker image, but we have to instantiate the entire
+    # stack to get the orchestrator to run pipelines.
+    image_hash = generate_image_hash(dockerfile=dockerfile)
+    logger.info(
+        "Building runner image %s for dockerfile:\n%s", image_hash, dockerfile
+    )
 
+    def _task() -> None:
         runner_image = workload_manager().build_and_push_image(
             workload_id=new_deployment.id,
             dockerfile=dockerfile,
@@ -322,10 +325,19 @@ def generate_dockerfile(
         pypi_requirements_string = " ".join(
             [f"'{r}'" for r in pypi_requirements]
         )
-        lines.append(
-            f"RUN pip install --default-timeout=60 --no-cache-dir "
-            f"{pypi_requirements_string}"
-        )
+
+        if handle_bool_env_var(
+            ENV_ZENML_RUNNER_IMAGE_DISABLE_UV, default=False
+        ):
+            lines.append(
+                f"RUN pip install --default-timeout=60 --no-cache-dir "
+                f"{pypi_requirements_string}"
+            )
+        else:
+            lines.append("RUN pip install uv")
+            lines.append(
+                f"RUN uv pip install --no-cache-dir {pypi_requirements_string}"
+            )
 
     return "\n".join(lines)
 
@@ -343,7 +355,8 @@ def deployment_request_from_template(
         user_id: ID of the user that is trying to run the template.
 
     Raises:
-        ValueError: If the run configuration is missing step parameters.
+        ValueError: If there are missing/extra step parameters in the run
+            configuration.
 
     Returns:
         The generated deployment request.
@@ -394,10 +407,17 @@ def deployment_request_from_template(
                 step_config_dict, update=update_dict
             )
 
-        if configured_parameters != required_parameters:
-            missing_parameters = required_parameters - configured_parameters
+        unknown_parameters = configured_parameters - required_parameters
+        if unknown_parameters:
+            raise ValueError(
+                "Run configuration contains the following unknown "
+                f"parameters for step {step.config.name}: {unknown_parameters}."
+            )
+
+        missing_parameters = required_parameters - configured_parameters
+        if missing_parameters:
             raise ValueError(
-                "Run configuration is missing missing the following required "
+                "Run configuration is missing the following required "
                 f"parameters for step {step.config.name}: {missing_parameters}."
             )
 

zenml/zen_stores/migrations/versions/6611d4bcc95b_add_hidden_option_for_templates.py

@@ -0,0 +1,47 @@
+"""Add hidden option for templates [6611d4bcc95b].
+
+Revision ID: 6611d4bcc95b
+Revises: 0.80.1
+Create Date: 2025-04-07 10:29:26.130057
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "6611d4bcc95b"
+down_revision = "0.80.1"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Upgrade database schema and/or data, creating a new revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("run_template", schema=None) as batch_op:
+        batch_op.add_column(sa.Column("hidden", sa.Boolean(), nullable=True))
+
+    op.execute(
+        """
+        UPDATE `run_template` 
+        SET hidden = 0
+        """
+    )
+
+    with op.batch_alter_table("run_template", schema=None) as batch_op:
+        batch_op.alter_column(
+            "hidden",
+            existing_type=sa.Boolean(),
+            nullable=False,
+        )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade database schema and/or data back to the previous revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("run_template", schema=None) as batch_op:
+        batch_op.drop_column("hidden")
+
+    # ### end Alembic commands ###

zenml/zen_stores/schemas/run_template_schemas.py

@@ -57,6 +57,11 @@ class RunTemplateSchema(NamedSchema, table=True):
         ),
     )
 
+    hidden: bool = Field(
+        default=False,
+        title="Whether the run template is hidden.",
+    )
+
     description: Optional[str] = Field(
         sa_column=Column(
             String(length=MEDIUMTEXT_MAX_LENGTH).with_variant(
@@ -166,6 +171,7 @@ class RunTemplateSchema(NamedSchema, table=True):
             project_id=request.project,
             name=request.name,
             description=request.description,
+            hidden=request.hidden,
             source_deployment_id=request.source_deployment_id,
         )
 
@@ -221,6 +227,7 @@ class RunTemplateSchema(NamedSchema, table=True):
             created=self.created,
             updated=self.updated,
             runnable=runnable,
+            hidden=self.hidden,
             latest_run_id=latest_run.id if latest_run else None,
             latest_run_status=latest_run.status if latest_run else None,
         )

{zenml_nightly-0.80.1.dev20250407.dist-info → zenml_nightly-0.80.1.dev20250409.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.80.1.dev20250407
+Version: 0.80.1.dev20250409
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -89,7 +89,7 @@ Requires-Dist: orjson (>=3.10.0,<3.11.0) ; extra == "server"
 Requires-Dist: packaging (>=24.1)
 Requires-Dist: passlib[bcrypt] (>=1.7.4,<1.8.0)
 Requires-Dist: psutil (>=5.0.0)
-Requires-Dist: pydantic (>=2.8,<2.9)
+Requires-Dist: pydantic (>=2.0,<2.11.2)
 Requires-Dist: pydantic-settings
 Requires-Dist: pyjwt[crypto] (==2.7.*) ; extra == "server"
 Requires-Dist: pyment (>=0.3.3,<0.4.0) ; extra == "dev"