From b0380f98bcd5f0015ce15a4c66c259d876abcc8c Mon Sep 17 00:00:00 2001 From: Chenqqian Zhang <100290172+Chengqian-Zhang@users.noreply.github.com> Date: Tue, 4 Jun 2024 18:27:51 +0800 Subject: [PATCH] Update deepmd/dpmodel/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/dpmodel/descriptor/se_atten_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/dpmodel/descriptor/se_atten_v2.py b/deepmd/dpmodel/descriptor/se_atten_v2.py index 8044e10eca..1375d2265f 100644 --- a/deepmd/dpmodel/descriptor/se_atten_v2.py +++ b/deepmd/dpmodel/descriptor/se_atten_v2.py @@ -155,7 +155,7 @@ def serialize(self) -> dict: return data @classmethod - def deserialize(cls, data: dict) -> "DescrptDPA1": + def deserialize(cls, data: dict) -> "DescrptSeAttenV2": """Deserialize from dict.""" data = data.copy() check_version_compatibility(data.pop("@version"), 1, 1)