diff --git a/deepmd/dpmodel/descriptor/__init__.py b/deepmd/dpmodel/descriptor/__init__.py
index 96b36a3f4a..5c3987e1c5 100644
--- 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
index 0ad690fd69..c247ccc573 100644
--- 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
\ No newline at end of file
+        return obj