zenml-nightly 0.58.2.dev20240614__py3-none-any.whl → 0.58.2.dev20240622__py3-none-any.whl

zenml/zen_stores/schemas/code_repository_schemas.py

@@ -90,7 +90,7 @@ class CodeRepositorySchema(NamedSchema, table=True):
             workspace_id=request.workspace,
             user_id=request.user,
             config=json.dumps(request.config),
-            source=request.source.json(),
+            source=request.source.model_dump_json(),
             description=request.description,
             logo_url=request.logo_url,
         )

zenml/zen_stores/schemas/component_schemas.py

@@ -49,7 +49,7 @@ class StackComponentSchema(NamedSchema, table=True):
 
     __tablename__ = "stack_component"
 
-    type: StackComponentType
+    type: str
     flavor: str
     configuration: bytes
     labels: Optional[bytes]
@@ -116,7 +116,7 @@ class StackComponentSchema(NamedSchema, table=True):
         Returns:
             The updated `StackComponentSchema`.
         """
-        for field, value in component_update.dict(
+        for field, value in component_update.model_dump(
             exclude_unset=True, exclude={"workspace", "user", "connector"}
         ).items():
             if field == "configuration":
@@ -127,6 +127,11 @@ class StackComponentSchema(NamedSchema, table=True):
                 self.labels = base64.b64encode(
                     json.dumps(component_update.labels).encode("utf-8")
                 )
+            elif field == "type":
+                component_type = component_update.type
+
+                if component_type is not None:
+                    self.type = component_type
             else:
                 setattr(self, field, value)
 
@@ -151,7 +156,7 @@ class StackComponentSchema(NamedSchema, table=True):
             A `ComponentModel`
         """
         body = ComponentResponseBody(
-            type=self.type,
+            type=StackComponentType(self.type),
             flavor=self.flavor,
             user=self.user.to_model() if self.user else None,
             created=self.created,

zenml/zen_stores/schemas/device_schemas.py

@@ -44,7 +44,7 @@ class OAuthDeviceSchema(BaseSchema, table=True):
     client_id: UUID
     user_code: str
     device_code: str
-    status: OAuthDeviceStatus
+    status: str
     failed_auth_attempts: int = 0
     expires: Optional[datetime] = None
     last_login: Optional[datetime] = None
@@ -121,7 +121,7 @@ class OAuthDeviceSchema(BaseSchema, table=True):
                 client_id=request.client_id,
                 user_code=hashed_user_code,
                 device_code=hashed_device_code,
-                status=OAuthDeviceStatus.PENDING,
+                status=OAuthDeviceStatus.PENDING.value,
                 failed_auth_attempts=0,
                 expires=now + timedelta(seconds=request.expires_in),
                 os=request.os,
@@ -148,14 +148,16 @@ class OAuthDeviceSchema(BaseSchema, table=True):
         Returns:
             The updated `OAuthDeviceSchema`.
         """
-        for field, value in device_update.dict(exclude_none=True).items():
+        for field, value in device_update.model_dump(
+            exclude_none=True
+        ).items():
             if hasattr(self, field):
                 setattr(self, field, value)
 
         if device_update.locked is True:
-            self.status = OAuthDeviceStatus.LOCKED
+            self.status = OAuthDeviceStatus.LOCKED.value
         elif device_update.locked is False:
-            self.status = OAuthDeviceStatus.ACTIVE
+            self.status = OAuthDeviceStatus.ACTIVE.value
 
         self.updated = datetime.utcnow()
         return self
@@ -233,7 +235,7 @@ class OAuthDeviceSchema(BaseSchema, table=True):
             client_id=self.client_id,
             expires=self.expires,
             trusted_device=self.trusted_device,
-            status=self.status,
+            status=OAuthDeviceStatus(self.status),
             os=self.os,
             ip_address=self.ip_address,
             hostname=self.hostname,

zenml/zen_stores/schemas/event_source_schemas.py

@@ -19,10 +19,8 @@ from datetime import datetime
 from typing import TYPE_CHECKING, Any, List, Optional, cast
 from uuid import UUID
 
-from pydantic import Field
-from pydantic.json import pydantic_encoder
 from sqlalchemy import TEXT, Column
-from sqlmodel import Relationship
+from sqlmodel import Field, Relationship
 
 from zenml import EventSourceResponseMetadata
 from zenml.models import (
@@ -33,6 +31,7 @@ from zenml.models import (
     EventSourceUpdate,
     Page,
 )
+from zenml.utils.json_utils import pydantic_encoder
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
 from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
 from zenml.zen_stores.schemas.user_schemas import UserSchema
@@ -174,7 +173,7 @@ class EventSourceSchema(NamedSchema, table=True):
         Returns:
             The updated `EventSourceSchema`.
         """
-        for field, value in update.dict(
+        for field, value in update.model_dump(
             exclude_unset=True, exclude_none=True
         ).items():
             if field == "configuration":

zenml/zen_stores/schemas/flavor_schemas.py

@@ -46,7 +46,7 @@ class FlavorSchema(NamedSchema, table=True):
 
     __tablename__ = "flavor"
 
-    type: StackComponentType
+    type: str
     source: str
     config_schema: str = Field(sa_column=Column(TEXT, nullable=False))
     integration: Optional[str] = Field(default="")
@@ -93,11 +93,13 @@ class FlavorSchema(NamedSchema, table=True):
         Returns:
             The updated `FlavorSchema`.
         """
-        for field, value in flavor_update.dict(
+        for field, value in flavor_update.model_dump(
             exclude_unset=True, exclude={"workspace", "user"}
         ).items():
             if field == "config_schema":
                 setattr(self, field, json.dumps(value))
+            elif field == "type":
+                setattr(self, field, value.value)
             else:
                 setattr(self, field, value)
 
@@ -123,7 +125,7 @@ class FlavorSchema(NamedSchema, table=True):
         """
         body = FlavorResponseBody(
             user=self.user.to_model() if self.user else None,
-            type=self.type,
+            type=StackComponentType(self.type),
             integration=self.integration,
             logo_url=self.logo_url,
             created=self.created,

zenml/zen_stores/schemas/model_schemas.py

@@ -17,6 +17,7 @@ from datetime import datetime
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, cast
 from uuid import UUID
 
+from pydantic import ConfigDict
 from sqlalchemy import BOOLEAN, INTEGER, TEXT, Column
 from sqlmodel import Field, Relationship
 
@@ -206,7 +207,7 @@ class ModelSchema(NamedSchema, table=True):
         Returns:
             The updated `ModelSchema`.
         """
-        for field, value in model_update.dict(
+        for field, value in model_update.model_dump(
             exclude_unset=True, exclude_none=True
         ).items():
             setattr(self, field, value)
@@ -286,6 +287,14 @@ class ModelVersionSchema(NamedSchema, table=True):
         ),
     )
 
+    # TODO: In Pydantic v2, the `model_` is a protected namespaces for all
+    #  fields defined under base models. If not handled, this raises a warning.
+    #  It is possible to suppress this warning message with the following
+    #  configuration, however the ultimate solution is to rename these fields.
+    #  Even though they do not cause any problems right now, if we are not
+    #  careful we might overwrite some fields protected by pydantic.
+    model_config = ConfigDict(protected_namespaces=())  # type: ignore[assignment]
+
     @classmethod
     def from_request(
         cls, model_version_request: ModelVersionRequest
@@ -498,6 +507,14 @@ class ModelVersionArtifactSchema(BaseSchema, table=True):
         sa_column=Column(BOOLEAN, nullable=True)
     )
 
+    # TODO: In Pydantic v2, the `model_` is a protected namespaces for all
+    #  fields defined under base models. If not handled, this raises a warning.
+    #  It is possible to suppress this warning message with the following
+    #  configuration, however the ultimate solution is to rename these fields.
+    #  Even though they do not cause any problems right now, if we are not
+    #  careful we might overwrite some fields protected by pydantic.
+    model_config = ConfigDict(protected_namespaces=())  # type: ignore[assignment]
+
     @classmethod
     def from_request(
         cls,
@@ -614,6 +631,14 @@ class ModelVersionPipelineRunSchema(BaseSchema, table=True):
         back_populates="model_versions_pipeline_runs_links"
     )
 
+    # TODO: In Pydantic v2, the `model_` is a protected namespaces for all
+    #  fields defined under base models. If not handled, this raises a warning.
+    #  It is possible to suppress this warning message with the following
+    #  configuration, however the ultimate solution is to rename these fields.
+    #  Even though they do not cause any problems right now, if we are not
+    #  careful we might overwrite some fields protected by pydantic.
+    model_config = ConfigDict(protected_namespaces=())  # type: ignore[assignment]
+
     @classmethod
     def from_request(
         cls,

zenml/zen_stores/schemas/pipeline_build_schemas.py

@@ -17,7 +17,6 @@ import json
 from typing import Any, Optional
 from uuid import UUID
 
-from pydantic.json import pydantic_encoder
 from sqlalchemy import Column, String
 from sqlalchemy.dialects.mysql import MEDIUMTEXT
 from sqlmodel import Field, Relationship
@@ -29,6 +28,7 @@ from zenml.models import (
     PipelineBuildResponseBody,
     PipelineBuildResponseMetadata,
 )
+from zenml.utils.json_utils import pydantic_encoder
 from zenml.zen_stores.schemas.base_schemas import BaseSchema
 from zenml.zen_stores.schemas.pipeline_schemas import PipelineSchema
 from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field

zenml/zen_stores/schemas/pipeline_deployment_schemas.py

@@ -17,7 +17,6 @@ import json
 from typing import TYPE_CHECKING, Any, List, Optional
 from uuid import UUID
 
-from pydantic.json import pydantic_encoder
 from sqlalchemy import TEXT, Column, String
 from sqlalchemy.dialects.mysql import MEDIUMTEXT
 from sqlmodel import Field, Relationship
@@ -31,6 +30,7 @@ from zenml.models import (
     PipelineDeploymentResponseBody,
     PipelineDeploymentResponseMetadata,
 )
+from zenml.utils.json_utils import pydantic_encoder
 from zenml.zen_stores.schemas.base_schemas import BaseSchema
 from zenml.zen_stores.schemas.code_repository_schemas import (
     CodeReferenceSchema,
@@ -178,7 +178,7 @@ class PipelineDeploymentSchema(BaseSchema, table=True):
             schedule_id=request.schedule,
             code_reference_id=code_reference_id,
             run_name_template=request.run_name_template,
-            pipeline_configuration=request.pipeline_configuration.json(),
+            pipeline_configuration=request.pipeline_configuration.model_dump_json(),
             step_configurations=json.dumps(
                 request.step_configurations,
                 sort_keys=False,
@@ -206,12 +206,12 @@ class PipelineDeploymentSchema(BaseSchema, table=True):
         Returns:
             The created `PipelineDeploymentResponse`.
         """
-        pipeline_configuration = PipelineConfiguration.parse_raw(
+        pipeline_configuration = PipelineConfiguration.model_validate_json(
             self.pipeline_configuration
         )
         step_configurations = json.loads(self.step_configurations)
         for s, c in step_configurations.items():
-            step_configurations[s] = Step.parse_obj(c)
+            step_configurations[s] = Step.model_validate(c)
 
         body = PipelineDeploymentResponseBody(
             user=self.user.to_model() if self.user else None,

zenml/zen_stores/schemas/pipeline_run_schemas.py

@@ -70,7 +70,7 @@ class PipelineRunSchema(NamedSchema, table=True):
     orchestrator_run_id: Optional[str] = Field(nullable=True)
     start_time: Optional[datetime] = Field(nullable=True)
     end_time: Optional[datetime] = Field(nullable=True, default=None)
-    status: ExecutionStatus = Field(nullable=False)
+    status: str = Field(nullable=False)
     orchestrator_environment: Optional[str] = Field(
         sa_column=Column(TEXT, nullable=True)
     )
@@ -209,7 +209,7 @@ class PipelineRunSchema(NamedSchema, table=True):
             orchestrator_run_id=request.orchestrator_run_id,
             orchestrator_environment=orchestrator_environment,
             start_time=request.start_time,
-            status=request.status,
+            status=request.status.value,
             pipeline_id=request.pipeline,
             deployment_id=request.deployment,
             trigger_execution_id=request.trigger_execution_id,
@@ -259,7 +259,7 @@ class PipelineRunSchema(NamedSchema, table=True):
             code_reference = deployment.code_reference
 
         elif self.pipeline_configuration is not None:
-            config = PipelineConfiguration.parse_raw(
+            config = PipelineConfiguration.model_validate_json(
                 self.pipeline_configuration
             )
             client_environment = (
@@ -283,7 +283,7 @@ class PipelineRunSchema(NamedSchema, table=True):
 
         body = PipelineRunResponseBody(
             user=self.user.to_model() if self.user else None,
-            status=self.status,
+            status=ExecutionStatus(self.status),
             stack=stack,
             pipeline=pipeline,
             build=build,
@@ -339,7 +339,7 @@ class PipelineRunSchema(NamedSchema, table=True):
             The updated `PipelineRunSchema`.
         """
         if run_update.status:
-            self.status = run_update.status
+            self.status = run_update.status.value
             self.end_time = run_update.end_time
 
         self.updated = datetime.utcnow()
@@ -384,7 +384,7 @@ class PipelineRunSchema(NamedSchema, table=True):
 
         self.orchestrator_run_id = request.orchestrator_run_id
         self.orchestrator_environment = orchestrator_environment
-        self.status = request.status
+        self.status = request.status.value
 
         self.updated = datetime.utcnow()
 

zenml/zen_stores/schemas/pipeline_schemas.py

@@ -13,6 +13,7 @@
 #  permissions and limitations under the License.
 """SQL Model Implementations for Pipelines and Pipeline Runs."""
 
+import json
 from datetime import datetime
 from typing import TYPE_CHECKING, Any, List, Optional
 from uuid import UUID
@@ -117,7 +118,9 @@ class PipelineSchema(NamedSchema, table=True):
             workspace_id=pipeline_request.workspace,
             user_id=pipeline_request.user,
             docstring=pipeline_request.docstring,
-            spec=pipeline_request.spec.json(sort_keys=True),
+            spec=json.dumps(
+                pipeline_request.spec.model_dump(mode="json"), sort_keys=True
+            ),
         )
 
     def to_model(
@@ -150,7 +153,7 @@ class PipelineSchema(NamedSchema, table=True):
             metadata = PipelineResponseMetadata(
                 workspace=self.workspace.to_model(),
                 version_hash=self.version_hash,
-                spec=PipelineSpec.parse_raw(self.spec),
+                spec=PipelineSpec.model_validate_json(self.spec),
                 docstring=self.docstring,
             )
 

zenml/zen_stores/schemas/run_metadata_schemas.py

@@ -109,7 +109,7 @@ class RunMetadataSchema(BaseSchema, table=True):
 
     key: str
     value: str = Field(sa_column=Column(TEXT, nullable=False))
-    type: MetadataTypeEnum
+    type: str
 
     def to_model(
         self,
@@ -134,7 +134,7 @@ class RunMetadataSchema(BaseSchema, table=True):
             created=self.created,
             updated=self.updated,
             value=json.loads(self.value),
-            type=self.type,
+            type=MetadataTypeEnum(self.type),
         )
         metadata = None
         if include_metadata:

zenml/zen_stores/schemas/secret_schemas.py

@@ -55,7 +55,7 @@ class SecretSchema(NamedSchema, table=True):
 
     __tablename__ = "secret"
 
-    scope: SecretScope
+    scope: str
 
     values: Optional[bytes] = Field(sa_column=Column(TEXT, nullable=True))
 
@@ -177,7 +177,7 @@ class SecretSchema(NamedSchema, table=True):
         assert secret.user is not None, "User must be set for secret creation."
         return cls(
             name=secret.name,
-            scope=secret.scope,
+            scope=secret.scope.value,
             workspace_id=secret.workspace,
             user_id=secret.user,
             # Don't store secret values implicitly in the secret. The
@@ -201,10 +201,13 @@ class SecretSchema(NamedSchema, table=True):
         # Don't update the secret values implicitly in the secret. The
         # SQL secret store will call `set_secret_values` to update the
         # values separately if SQL is used as the secrets store.
-        for field, value in secret_update.dict(
+        for field, value in secret_update.model_dump(
             exclude_unset=True, exclude={"workspace", "user", "values"}
         ).items():
-            setattr(self, field, value)
+            if field == "scope":
+                setattr(self, field, value.value)
+            else:
+                setattr(self, field, value)
 
         self.updated = datetime.utcnow()
         return self
@@ -239,7 +242,7 @@ class SecretSchema(NamedSchema, table=True):
             user=self.user.to_model() if self.user else None,
             created=self.created,
             updated=self.updated,
-            scope=self.scope,
+            scope=SecretScope(self.scope),
         )
         return SecretResponse(
             id=self.id,

zenml/zen_stores/schemas/server_settings_schemas.py

@@ -56,7 +56,9 @@ class ServerSettingsSchema(SQLModel, table=True):
         Returns:
             The updated `ServerSettingsSchema`.
         """
-        for field, value in settings_update.dict(exclude_unset=True).items():
+        for field, value in settings_update.model_dump(
+            exclude_unset=True
+        ).items():
             if field == "onboarding_state":
                 if value is not None:
                     self.onboarding_state = json.dumps(value)

zenml/zen_stores/schemas/service_connector_schemas.py

@@ -186,7 +186,7 @@ class ServiceConnectorSchema(NamedSchema, table=True):
         Returns:
             The updated `ServiceConnectorSchema`.
         """
-        for field, value in connector_update.dict(
+        for field, value in connector_update.model_dump(
             exclude_unset=False,
             exclude={"workspace", "user", "secrets"},
         ).items():

zenml/zen_stores/schemas/service_schemas.py

@@ -19,6 +19,7 @@ from datetime import datetime
 from typing import Any, Optional
 from uuid import UUID
 
+from pydantic import ConfigDict
 from sqlalchemy import TEXT, Column
 from sqlmodel import Field, Relationship
 
@@ -108,6 +109,14 @@ class ServiceSchema(NamedSchema, table=True):
         back_populates="services",
     )
 
+    # TODO: In Pydantic v2, the `model_` is a protected namespaces for all
+    #  fields defined under base models. If not handled, this raises a warning.
+    #  It is possible to suppress this warning message with the following
+    #  configuration, however the ultimate solution is to rename these fields.
+    #  Even though they do not cause any problems right now, if we are not
+    #  careful we might overwrite some fields protected by pydantic.
+    model_config = ConfigDict(protected_namespaces=())  # type: ignore[assignment]
+
     def to_model(
         self,
         include_metadata: bool = False,
@@ -181,7 +190,7 @@ class ServiceSchema(NamedSchema, table=True):
         Returns:
             The updated `ServiceSchema`.
         """
-        for field, value in update.dict(
+        for field, value in update.model_dump(
             exclude_unset=True, exclude_none=True
         ).items():
             if field == "labels":
@@ -221,7 +230,7 @@ class ServiceSchema(NamedSchema, table=True):
             workspace_id=service_request.workspace,
             user_id=service_request.user,
             service_source=service_request.service_source,
-            service_type=service_request.service_type.json(),
+            service_type=service_request.service_type.model_dump_json(),
             type=service_request.service_type.type,
             flavor=service_request.service_type.flavor,
             admin_state=service_request.admin_state,

zenml/zen_stores/schemas/stack_schemas.py

@@ -119,7 +119,7 @@ class StackSchema(NamedSchema, table=True):
         Returns:
             The updated StackSchema.
         """
-        for field, value in stack_update.dict(
+        for field, value in stack_update.model_dump(
             exclude_unset=True, exclude={"workspace", "user"}
         ).items():
             if field == "components":

zenml/zen_stores/schemas/step_run_schemas.py

@@ -27,8 +27,6 @@ from zenml.constants import MEDIUMTEXT_MAX_LENGTH
 from zenml.enums import (
     ExecutionStatus,
     MetadataResourceTypes,
-    StepRunInputArtifactType,
-    StepRunOutputArtifactType,
 )
 from zenml.models import (
     StepRunRequest,
@@ -61,7 +59,7 @@ class StepRunSchema(NamedSchema, table=True):
     # Fields
     start_time: Optional[datetime] = Field(nullable=True)
     end_time: Optional[datetime] = Field(nullable=True)
-    status: ExecutionStatus = Field(nullable=False)
+    status: str = Field(nullable=False)
 
     docstring: Optional[str] = Field(sa_column=Column(TEXT, nullable=True))
     cache_key: Optional[str] = Field(nullable=True)
@@ -166,7 +164,7 @@ class StepRunSchema(NamedSchema, table=True):
             user_id=request.user,
             start_time=request.start_time,
             end_time=request.end_time,
-            status=request.status,
+            status=request.status.value,
             original_step_run_id=request.original_step_run_id,
             pipeline_run_id=request.pipeline_run_id,
             deployment_id=request.deployment,
@@ -215,11 +213,13 @@ class StepRunSchema(NamedSchema, table=True):
         }
 
         if self.deployment is not None:
-            full_step_config = Step.parse_obj(
+            full_step_config = Step.model_validate(
                 json.loads(self.deployment.step_configurations)[self.name]
             )
         elif self.step_configuration is not None:
-            full_step_config = Step.parse_raw(self.step_configuration)
+            full_step_config = Step.model_validate_json(
+                self.step_configuration
+            )
         else:
             raise RuntimeError(
                 "Step run model creation has failed. Each step run entry "
@@ -228,7 +228,7 @@ class StepRunSchema(NamedSchema, table=True):
 
         body = StepRunResponseBody(
             user=self.user.to_model() if self.user else None,
-            status=self.status,
+            status=ExecutionStatus(self.status),
             inputs=input_artifacts,
             outputs=output_artifacts,
             created=self.created,
@@ -282,11 +282,11 @@ class StepRunSchema(NamedSchema, table=True):
         Returns:
             The updated step run schema.
         """
-        for key, value in step_update.dict(
+        for key, value in step_update.model_dump(
             exclude_unset=True, exclude_none=True
         ).items():
             if key == "status":
-                self.status = value
+                self.status = value.value
             if key == "end_time":
                 self.end_time = value
 
@@ -328,7 +328,7 @@ class StepRunInputArtifactSchema(SQLModel, table=True):
 
     # Fields
     name: str = Field(nullable=False, primary_key=True)
-    type: StepRunInputArtifactType
+    type: str
 
     # Foreign keys
     step_id: UUID = build_foreign_key_field(
@@ -364,7 +364,7 @@ class StepRunOutputArtifactSchema(SQLModel, table=True):
 
     # Fields
     name: str
-    type: StepRunOutputArtifactType
+    type: str
 
     # Foreign keys
     step_id: UUID = build_foreign_key_field(

zenml/zen_stores/schemas/tag_schemas.py

@@ -107,8 +107,12 @@ class TagSchema(NamedSchema, table=True):
         Returns:
             The updated `TagSchema`.
         """
-        for field, value in update.dict(exclude_unset=True).items():
-            setattr(self, field, value)
+        for field, value in update.model_dump(exclude_unset=True).items():
+            if field == "color":
+                setattr(self, field, value.value)
+            else:
+                setattr(self, field, value)
+
         self.updated = datetime.utcnow()
         return self
 

zenml/zen_stores/schemas/trigger_schemas.py

@@ -19,7 +19,6 @@ from datetime import datetime
 from typing import Any, List, Optional, cast
 from uuid import UUID
 
-from pydantic.json import pydantic_encoder
 from sqlalchemy import TEXT, Column
 from sqlmodel import Field, Relationship
 
@@ -37,6 +36,7 @@ from zenml.models import (
     TriggerResponseResources,
     TriggerUpdate,
 )
+from zenml.utils.json_utils import pydantic_encoder
 from zenml.zen_stores.schemas.base_schemas import BaseSchema, NamedSchema
 from zenml.zen_stores.schemas.event_source_schemas import EventSourceSchema
 from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
@@ -121,7 +121,7 @@ class TriggerSchema(NamedSchema, table=True):
         Returns:
             The updated TriggerSchema.
         """
-        for field, value in trigger_update.dict(
+        for field, value in trigger_update.model_dump(
             exclude_unset=True,
             exclude_none=True,
         ).items():

zenml/zen_stores/schemas/user_schemas.py

@@ -210,7 +210,7 @@ class UserSchema(NamedSchema, table=True):
         Returns:
             The updated `UserSchema`.
         """
-        for field, value in user_update.dict(exclude_unset=True).items():
+        for field, value in user_update.model_dump(exclude_unset=True).items():
             if field == "old_password":
                 continue
 
@@ -241,7 +241,7 @@ class UserSchema(NamedSchema, table=True):
         Returns:
             The updated `UserSchema`.
         """
-        for field, value in service_account_update.dict(
+        for field, value in service_account_update.model_dump(
             exclude_none=True
         ).items():
             setattr(self, field, value)

zenml/zen_stores/schemas/workspace_schemas.py

@@ -172,7 +172,9 @@ class WorkspaceSchema(NamedSchema, table=True):
         Returns:
             The updated `WorkspaceSchema`.
         """
-        for field, value in workspace_update.dict(exclude_unset=True).items():
+        for field, value in workspace_update.model_dump(
+            exclude_unset=True
+        ).items():
             setattr(self, field, value)
 
         self.updated = datetime.utcnow()