Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jan 4, 2025
1 parent 6d38b94 commit 6bbced8
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions deepmd/pt/utils/stat.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,10 +153,9 @@ def finalize_stats(sys_stat):
pass
else:
global_element_counts[elem]["count"] += count
global_element_counts[elem]["indices"].append({
"sys_index": sys_index,
"frames": indices
})
global_element_counts[elem]["indices"].append(
{"sys_index": sys_index, "frames": indices}
)
if datasets[0].mixed_type and enable_element_completion:
for elem, data in global_element_counts.items():
indices_count = data["count"]
Expand Down

0 comments on commit 6bbced8

Please sign in to comment.