Skip to content

Commit 6ea306f

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 38a0273 commit 6ea306f

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

deepmd/pt/model/network/utils.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -136,8 +136,6 @@ def get_graph_index(
136136
eik2a_index = edge_index_ik[a_nlist_mask_3d]
137137

138138
edge_index_result = torch.stack([n2e_index, n_ext2e_index], dim=0)
139-
angle_index_result = torch.stack(
140-
[n2a_index, eij2a_index, eik2a_index], dim=0
141-
)
139+
angle_index_result = torch.stack([n2a_index, eij2a_index, eik2a_index], dim=0)
142140

143141
return edge_index_result, angle_index_result

0 commit comments

Comments
 (0)