From df1f4fbc63ed4104c1350b06e91a20e3283b6f67 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 1 Apr 2024 21:54:45 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 deepmd/tf/descriptor/se_atten.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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