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