Skip to content

Commit 5c7f1bf

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

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

deepmd/pt/model/model/ener_model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -60,9 +60,9 @@ def get_observed_type_list(self) -> list[str]:
6060
assert out_bias.size(0) == len(type_map), (
6161
"The out_bias shape does not match the type_map length."
6262
)
63-
bias_mask = torch.gt(torch.abs(out_bias), 1e-6).any(
64-
dim=-1
65-
).detach().cpu() # 1e-6 for stability
63+
bias_mask = (
64+
torch.gt(torch.abs(out_bias), 1e-6).any(dim=-1).detach().cpu()
65+
) # 1e-6 for stability
6666

6767
observed_type_list: list[str] = []
6868
for i in range(len(type_map)):

0 commit comments

Comments
 (0)