diff --git a/deepmd/tf/descriptor/se_atten.py b/deepmd/tf/descriptor/se_atten.py
index 5553e2e3c4..82184dec02 100644
--- 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