diff --git a/deepmd/utils/out_stat.py b/deepmd/utils/out_stat.py
index 5a64138a40..d241b2f217 100644
--- a/deepmd/utils/out_stat.py
+++ b/deepmd/utils/out_stat.py
@@ -181,7 +181,9 @@ def compute_stats_do_not_distinguish_types(
         )
         output_std = np.std(output_redu, axis=0)
 
-        computed_output_bias = computed_output_bias.reshape([natoms.shape[1]] + var_shape)  # noqa: RUF005
+        computed_output_bias = computed_output_bias.reshape(
+            [natoms.shape[1]] + var_shape
+        )  # noqa: RUF005
         output_std = output_std.reshape(var_shape)
         output_std = np.tile(output_std, (computed_output_bias.shape[0], 1))
     else: