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

for more information, see https://pre-commit.ci
deepmodeling · Apr 1, 2024 · 56e6cac · 56e6cac
1 parent 9dfa85a
commit 56e6cac
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
@@ -682,7 +682,9 @@ def _pass_filter(
                 self.nei_type_vec,  # extra input for atten
             )
             if self.smooth:
-                inputs_i = tf.where(tf.cast(mask, tf.bool), inputs_i, self.avg_looked_up)
+                inputs_i = tf.where(
+                    tf.cast(mask, tf.bool), inputs_i, self.avg_looked_up
+                )
             else:
                 inputs_i *= mask
         if nvnmd_cfg.enable and nvnmd_cfg.quantize_descriptor: