From 66158373d168eeab8087301bc0fa01ce755441cb Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 23 Dec 2024 05:17:16 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- deepmd/utils/out_stat.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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: