From 33daa4bd3d47a15a495fc867f028bd8b324922a5 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 23 Oct 2024 09:11:00 +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 55d163ec63..8d101f151c 100644
--- a/deepmd/tf/descriptor/se_atten.py
+++ b/deepmd/tf/descriptor/se_atten.py
@@ -2288,7 +2288,9 @@ def enable_compression(
         tebd_suffix : str, optional
             Same as suffix.
         """
-        assert tebd_suffix == "", "DescrptDPA1Compat must use the same tebd_suffix as suffix!"
+        assert (
+            tebd_suffix == ""
+        ), "DescrptDPA1Compat must use the same tebd_suffix as suffix!"
         super().enable_compression(
             min_nbor_dist,
             graph,