From ef94c04d220209f0ad556f4f43c6b9cec7e6c48c Mon Sep 17 00:00:00 2001 From: Chenqqian Zhang <100290172+Chengqian-Zhang@users.noreply.github.com> Date: Tue, 4 Jun 2024 18:28:12 +0800 Subject: [PATCH] Update deepmd/pt/model/descriptor/se_atten_v2.py Co-authored-by: Duo <50307526+iProzd@users.noreply.github.com> Signed-off-by: Chenqqian Zhang <100290172+Chengqian-Zhang@users.noreply.github.com> --- deepmd/pt/model/descriptor/se_atten_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/pt/model/descriptor/se_atten_v2.py b/deepmd/pt/model/descriptor/se_atten_v2.py index 2a55af302d..3b350ded98 100644 --- a/deepmd/pt/model/descriptor/se_atten_v2.py +++ b/deepmd/pt/model/descriptor/se_atten_v2.py @@ -222,7 +222,7 @@ def serialize(self) -> dict: return data @classmethod - def deserialize(cls, data: dict) -> "DescrptDPA1": + def deserialize(cls, data: dict) -> "DescrptSeAttenV2": data = data.copy() check_version_compatibility(data.pop("@version"), 1, 1) data.pop("@class")