Skip to content

Commit 4a58ae2

Browse files
committed
Add custom serialization to drop name attribute
1 parent a37dd41 commit 4a58ae2

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

schema/BaseElement.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,8 @@
1-
from pydantic import BaseModel, ConfigDict
1+
from pydantic import (
2+
BaseModel,
3+
ConfigDict,
4+
model_serializer,
5+
)
26
from typing import Literal, Optional
37

48

@@ -14,3 +18,9 @@ class BaseElement(BaseModel):
1418

1519
# Unique element name
1620
name: Optional[str] = None
21+
22+
@model_serializer
23+
def _serialize(self):
24+
data = self.__dict__.copy()
25+
data.pop("name", None)
26+
return data

0 commit comments

Comments
 (0)