add from_dict back in
This commit is contained in:
1 parent
50767fed40
commit
f91c4f6b20
2 files changed
+6
-1
No files matched your search
@@ -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))
|
||||
|
||||
|
||||
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user