Skip to content

Commit 10a7424

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

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

deepmd/pd/model/descriptor/repflow_layer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -586,7 +586,9 @@ def optim_angle_update_dynamic(
586586
sub_node_update = paddle.matmul(node_ebd, sub_node)
587587
# n_angle * angle_dim
588588
sub_node_update = paddle.index_select(
589-
sub_node_update.reshape((nf * nloc, sub_node_update.shape[-1])), n2a_index, 0
589+
sub_node_update.reshape((nf * nloc, sub_node_update.shape[-1])),
590+
n2a_index,
591+
0,
590592
)
591593

592594
# n_edge * angle_dim

0 commit comments

Comments
 (0)