From dce0618cc40bbfeef15733ae1f978be945082f51 Mon Sep 17 00:00:00 2001
From: nahso <topcloud@foxmail.com>
Date: Mon, 25 Sep 2023 13:57:24 +0800
Subject: [PATCH] pre-commit

---
 deepmd/descriptor/__init__.py | 6 +++---
 deepmd/descriptor/se_a.py     | 4 +---
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/deepmd/descriptor/__init__.py b/deepmd/descriptor/__init__.py
index fa35e05888..6ef6c50da5 100644
--- a/deepmd/descriptor/__init__.py
+++ b/deepmd/descriptor/__init__.py
@@ -14,6 +14,9 @@
 from .se_a_ebd import (
     DescrptSeAEbd,
 )
+from .se_a_ebd_v2 import (
+    DescrptSeAEbdV2,
+)
 from .se_a_ef import (
     DescrptSeAEf,
     DescrptSeAEfLower,
@@ -21,9 +24,6 @@
 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
index 7f2db3bc83..8f0051cd4e 100644
--- 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 = []