fix bugs

deepmodeling · Mar 2, 2024 · 4d21d0d · 4d21d0d
1 parent 17d7128
commit 4d21d0d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/deepmd/dpmodel/atomic_model/base_atomic_model.py b/deepmd/dpmodel/atomic_model/base_atomic_model.py
@@ -61,15 +61,16 @@ def atomic_output_def(self) -> FittingOutputDef:
         else:
             old_list = list(old_def.get_data().values())
             return FittingOutputDef(
-                old_list.append(
+                old_list  # noqa:RUF005
+                + [
                     OutputVariableDef(
                         name="mask",
                         shape=[1],
                         reduciable=False,
                         r_differentiable=False,
                         c_differentiable=False,
                     )
-                )
+                ]
             )
 
     def forward_common_atomic(

diff --git a/deepmd/pt/model/atomic_model/base_atomic_model.py b/deepmd/pt/model/atomic_model/base_atomic_model.py
@@ -71,15 +71,16 @@ def atomic_output_def(self) -> FittingOutputDef:
         else:
             old_list = list(old_def.get_data().values())
             return FittingOutputDef(
-                old_list.append(
+                old_list  # noqa:RUF005
+                + [
                     OutputVariableDef(
                         name="mask",
                         shape=[1],
                         reduciable=False,
                         r_differentiable=False,
                         c_differentiable=False,
                     )
-                )
+                ]
             )
 
     def forward_common_atomic(