Update deepmd/pt/model/descriptor/se_atten_v2.py

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