Update deepmd/dpmodel/descriptor/se_atten_v2.py

Co-authored-by: Duo <[email protected]> Signed-off-by: Chenqqian Zhang <[email protected]>
deepmodeling · Jun 4, 2024 · b0380f9 · b0380f9
1 parent 8ee1125
commit b0380f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)