add from_dict back in

This commit is contained in:
stuart committed 2026-02-25 14:13:00 +01:00
1 parent 50767fed40
commit f91c4f6b20
2 files changed
+6 -1

No files matched your search

+5
View File
@@ -29,4 +29,9 @@ Tensor = Annotated[torch.Tensor, TensorPydantic]
class Embedding(BaseModel):
data: Tensor
@classmethod
def from_dict(cls, data: dict) -> "Embedding":
return cls(data=torch.tensor(data["data"], dtype=torch.float32))
+1 -1
View File
@@ -17,7 +17,7 @@ class Lexeme(BaseModel,ABC):
return cls(
id=data["id"],
content=data["content"],
embedding=Embedding(data["embedding"]),
embedding=Embedding.from_dict(data["embedding"]),
original_content=data["original_content"],
)
def __new__(cls, *args, **kwargs):