Skip to content

Commit 61a29aa

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

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

deepmd/pt/model/descriptor/repflows.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -503,7 +503,10 @@ def forward(
503503
if self.use_dynamic_sel:
504504
# get graph index
505505
edge_index, angle_index = get_graph_index(
506-
nlist, nlist_mask, a_nlist_mask, nall,
506+
nlist,
507+
nlist_mask,
508+
a_nlist_mask,
509+
nall,
507510
use_loc_mapping=self.use_loc_mapping,
508511
)
509512
# flat all the tensors

0 commit comments

Comments
 (0)