From 7893fa2d3ea753736bbf608ba541eab7531b08f8 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Fri, 12 Jan 2024 03:37:53 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 deepmd/descriptor/se_a_mask.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/deepmd/descriptor/se_a_mask.py b/deepmd/descriptor/se_a_mask.py
index 346e72c908..e32492c9df 100644
--- a/deepmd/descriptor/se_a_mask.py
+++ b/deepmd/descriptor/se_a_mask.py
@@ -536,7 +536,7 @@ def _filter_lower(
         is_exclude=False,
     ):
         """Input env matrix, returns R.G."""
-        outputs_size = [1] + self.filter_neuron
+        outputs_size = [1, *self.filter_neuron]
         # cut-out inputs
         # with natom x (nei_type_i x 4)
         inputs_i = paddle.slice(
@@ -663,13 +663,11 @@ def _filter(
         nframes = 1
         # natom x (nei x 4)
         shape = inputs.shape
-        outputs_size = [1] + self.filter_neuron
+        outputs_size = [1, *self.filter_neuron]
         outputs_size_2 = self.n_axis_neuron  # 16
         all_excluded = all(
-            [
-                (type_input, type_i) in self.exclude_types  #  set()
+            (type_input, type_i) in self.exclude_types  #  set()
                 for type_i in range(self.ntypes)
-            ]
         )  # False
         if all_excluded:
             # all types are excluded so result and qmat should be zeros