add from_dict in noembeddingphrase

This commit is contained in:
stuart committed 2026-02-25 15:46:21 +01:00
1 parent d51322d34f
commit 15afc9d481
1 file changed
+9
+9
View File
@@ -12,6 +12,13 @@ class NoEmbeddingPhrase(Lexeme):
#initialised with empty embedding
super().__init__(id=id, content=content, embedding=Embedding(data=torch.tensor([])), original_content=original_content)
@classmethod
def from_dict(cls, data: dict) -> "NoEmbeddingPhrase":
return cls(
id=data["id"],
content=data["content"],
original_content=data["original_content"],
)
class Phrase(Lexeme):
@@ -20,6 +27,7 @@ class Phrase(Lexeme):
return cls(
id=data["id"],
content=data["content"],
original_content=data["original_content"],
embedding=Embedding.from_dict(data["embedding"]),
)
@@ -30,5 +38,6 @@ class Phrase(Lexeme):
return cls(
id=phrase.id,
content=phrase.content,
original_content=phrase.original_content,
embedding=embedding,
)