[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
deepmodeling · Mar 14, 2024 · e5b6cf0 · e5b6cf0
1 parent a61f462
commit e5b6cf0
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/deepmd/tf/fit/dos.py b/deepmd/tf/fit/dos.py
@@ -528,7 +528,11 @@ def build(
 
                 final_layer = tf.reshape(
                     final_layer,
-                    [tf.shape(inputs)[0] ,natoms[2 + type_i],self.numb_dos, ],
+                    [
+                        tf.shape(inputs)[0],
+                        natoms[2 + type_i],
+                        self.numb_dos,
+                    ],
                 )
                 outs_list.append(final_layer)
                 start_index += natoms[2 + type_i]
@@ -557,7 +561,12 @@ def build(
             )
 
             outs = tf.reshape(
-                final_layer, [tf.shape(inputs)[0] ,natoms[2 + type_i],self.numb_dos, ],
+                final_layer,
+                [
+                    tf.shape(inputs)[0],
+                    natoms[2 + type_i],
+                    self.numb_dos,
+                ],
             )
         # add bias
         # self.atom_ener_before = outs