diff --git a/deepmd/pt/model/atomic_model/linear_atomic_model.py b/deepmd/pt/model/atomic_model/linear_atomic_model.py
index 7fe23523a7..87b58388f2 100644
--- a/deepmd/pt/model/atomic_model/linear_atomic_model.py
+++ b/deepmd/pt/model/atomic_model/linear_atomic_model.py
@@ -193,13 +193,13 @@ def forward_atomic(
             # apply bias to each individual model
             ener_list.append(
                 model.forward_common_atomic(
-                extended_coord,
-                mapping[extended_atype],
-                nlists_[i],
-                mapping,
-                fparam,
-                aparam,
-            )["energy"]
+                    extended_coord,
+                    mapping[extended_atype],
+                    nlists_[i],
+                    mapping,
+                    fparam,
+                    aparam,
+                )["energy"]
             )
         weights = self._compute_weight(extended_coord, extended_atype, nlists_)