Skip to content

Commit 5acb937

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent f513b7b commit 5acb937

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

deepmd/dpmodel/model/transform_output.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,9 @@ def communicate_extended_output(
100100
if vdef.r_differentiable:
101101
if model_ret[kk_derv_r] is not None:
102102
derv_r_ext_dims = list(vdef.shape) + [3] # noqa:RUF005
103-
mapping = xp.reshape(mapping, tuple(mldims + [1] * len(derv_r_ext_dims)))
103+
mapping = xp.reshape(
104+
mapping, tuple(mldims + [1] * len(derv_r_ext_dims))
105+
)
104106
mapping = xp.tile(mapping, [1] * len(mldims) + derv_r_ext_dims)
105107
force = xp.zeros(vldims + derv_r_ext_dims, dtype=vv.dtype)
106108
force = xp_scatter_sum(

0 commit comments

Comments
 (0)