pre-commit

deepmodeling · Sep 25, 2023 · dce0618 · dce0618
1 parent 885816b
commit dce0618
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/deepmd/descriptor/__init__.py b/deepmd/descriptor/__init__.py
@@ -14,16 +14,16 @@
 from .se_a_ebd import (
     DescrptSeAEbd,
 )
+from .se_a_ebd_v2 import (
+    DescrptSeAEbdV2,
+)
 from .se_a_ef import (
     DescrptSeAEf,
     DescrptSeAEfLower,
 )
 from .se_a_mask import (
     DescrptSeAMask,
 )
-from .se_a_ebd_v2 import (
-    DescrptSeAEbdV2,
-)
 from .se_atten import (
     DescrptSeAtten,
 )

diff --git a/deepmd/descriptor/se_a.py b/deepmd/descriptor/se_a.py
@@ -744,9 +744,7 @@ def _pass_filter(
         else:
             type_embedding = None
         if self.stripped_type_embedding and type_embedding is None:
-            raise RuntimeError(
-                "type_embedding is required for se_a_tebd_v2 model."
-            )
+            raise RuntimeError("type_embedding is required for se_a_tebd_v2 model.")
         start_index = 0
         inputs = tf.reshape(inputs, [-1, natoms[0], self.ndescrpt])
         output = []