diff --git a/src/zenml/models/v2/core/pipeline_run.py b/src/zenml/models/v2/core/pipeline_run.py index 0212eb8451b..58e5a7fde6d 100644 --- a/src/zenml/models/v2/core/pipeline_run.py +++ b/src/zenml/models/v2/core/pipeline_run.py @@ -121,6 +121,8 @@ class PipelineRunRequest(WorkspaceScopedRequest): default=None, ) + model_config = ConfigDict(protected_namespaces=()) + # ------------------ Update Model ------------------ @@ -143,6 +145,8 @@ class PipelineRunUpdate(BaseModel): default=None, title="Tags to remove from the pipeline run." ) + model_config = ConfigDict(protected_namespaces=()) + # ------------------ Response Model ------------------ @@ -180,6 +184,8 @@ class PipelineRunResponseBody(WorkspaceScopedResponseBody): default=None, ) + model_config = ConfigDict(protected_namespaces=()) + class PipelineRunResponseMetadata(WorkspaceScopedResponseMetadata): """Response metadata for pipeline runs.""" diff --git a/src/zenml/models/v2/core/step_run.py b/src/zenml/models/v2/core/step_run.py index d49b6893447..2030c6c86ba 100644 --- a/src/zenml/models/v2/core/step_run.py +++ b/src/zenml/models/v2/core/step_run.py @@ -114,6 +114,8 @@ class StepRunRequest(WorkspaceScopedRequest): default=None, ) + model_config = ConfigDict(protected_namespaces=()) + # ------------------ Update Model ------------------ @@ -146,6 +148,7 @@ class StepRunUpdate(BaseModel): "configured by this step run explicitly.", default=None, ) + model_config = ConfigDict(protected_namespaces=()) # ------------------ Response Model ------------------ @@ -166,6 +169,7 @@ class StepRunResponseBody(WorkspaceScopedResponseBody): "configured by this step run explicitly.", default=None, ) + model_config = ConfigDict(protected_namespaces=()) class StepRunResponseMetadata(WorkspaceScopedResponseMetadata): diff --git a/src/zenml/zen_stores/schemas/pipeline_run_schemas.py b/src/zenml/zen_stores/schemas/pipeline_run_schemas.py index 4c43a0dc34e..f1605e2648e 100644 --- a/src/zenml/zen_stores/schemas/pipeline_run_schemas.py +++ b/src/zenml/zen_stores/schemas/pipeline_run_schemas.py @@ -18,6 +18,7 @@ from typing import TYPE_CHECKING, Any, List, Optional from uuid import UUID +from pydantic import ConfigDict from sqlalchemy import UniqueConstraint from sqlmodel import TEXT, Column, Field, Relationship @@ -213,6 +214,8 @@ class PipelineRunSchema(NamedSchema, table=True): ), ) + model_config = ConfigDict(protected_namespaces=()) # type: ignore[assignment] + @classmethod def from_request( cls, request: "PipelineRunRequest" diff --git a/src/zenml/zen_stores/schemas/step_run_schemas.py b/src/zenml/zen_stores/schemas/step_run_schemas.py index e2c6d7c34c5..07812b26ec4 100644 --- a/src/zenml/zen_stores/schemas/step_run_schemas.py +++ b/src/zenml/zen_stores/schemas/step_run_schemas.py @@ -18,6 +18,7 @@ from typing import TYPE_CHECKING, Any, List, Optional from uuid import UUID +from pydantic import ConfigDict from sqlalchemy import TEXT, Column, String from sqlalchemy.dialects.mysql import MEDIUMTEXT from sqlmodel import Field, Relationship, SQLModel @@ -161,6 +162,8 @@ class StepRunSchema(NamedSchema, table=True): back_populates="step_runs", ) + model_config = ConfigDict(protected_namespaces=()) # type: ignore[assignment] + @classmethod def from_request(cls, request: StepRunRequest) -> "StepRunSchema": """Create a step run schema from a step run request model.