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

for more information, see https://pre-commit.ci
deepmodeling · Apr 1, 2024 · df1f4fb · df1f4fb
1 parent db0373e
commit df1f4fb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deepmd/tf/descriptor/se_atten.py b/deepmd/tf/descriptor/se_atten.py
@@ -686,7 +686,9 @@ def _pass_filter(
                     tf.cast(mask, tf.bool),
                     inputs_i,
                     # (nframes * nloc, 1) -> (nframes * nloc, ndescrpt)
-                    tf.tile(tf.reshape(self.avg_looked_up, [-1, 1]), [1, self.ndescrpt]),
+                    tf.tile(
+                        tf.reshape(self.avg_looked_up, [-1, 1]), [1, self.ndescrpt]
+                    ),
                 )
             else:
                 inputs_i *= mask