Silence pydantic protected namespace warnings

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."""

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):

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"

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.