From f91c4f6b20868ed58b8b0b64645e4bb0c6dc20d8 Mon Sep 17 00:00:00 2001 From: stuartgo Date: Wed, 25 Feb 2026 14:13:00 +0100 Subject: [PATCH] add from_dict back in --- anansyTypes/structures/embedding.py | 5 +++++ anansyTypes/structures/lexeme.py | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/anansyTypes/structures/embedding.py b/anansyTypes/structures/embedding.py index 33caa29..b396863 100644 --- a/anansyTypes/structures/embedding.py +++ b/anansyTypes/structures/embedding.py @@ -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)) + \ No newline at end of file diff --git a/anansyTypes/structures/lexeme.py b/anansyTypes/structures/lexeme.py index b891a53..25c1a10 100644 --- a/anansyTypes/structures/lexeme.py +++ b/anansyTypes/structures/lexeme.py @@ -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):