Merge branch '3831' of github.com:Chengqian-Zhang/deepmd-kit into 3831

deepmodeling · Jun 4, 2024 · d25993f · d25993f
2 parents 9d4d0e3 + d2fd8a2
commit d25993f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/deepmd/dpmodel/descriptor/__init__.py b/deepmd/dpmodel/descriptor/__init__.py
@@ -2,9 +2,6 @@
 from .dpa1 import (
     DescrptDPA1,
 )
-from .se_atten_v2 import (
-    DescrptSeAttenV2,
-)
 from .dpa2 import (
     DescrptDPA2,
 )
@@ -14,6 +11,9 @@
 from .make_base_descriptor import (
     make_base_descriptor,
 )
+from .se_atten_v2 import (
+    DescrptSeAttenV2,
+)
 from .se_e2_a import (
     DescrptSeA,
 )

diff --git a/deepmd/dpmodel/descriptor/se_atten_v2.py b/deepmd/dpmodel/descriptor/se_atten_v2.py
@@ -4,7 +4,7 @@
     List,
     Optional,
     Tuple,
-    Union,    
+    Union,
 )
 
 import numpy as np
@@ -177,4 +177,4 @@ def deserialize(cls, data: dict) -> "DescrptDPA1":
         obj.se_atten.dpa1_attention = NeighborGatedAttention.deserialize(
             attention_layers
         )
-        return obj
+        return obj