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 Apr 1, 2024
1 parent 6f8eff4 commit 3314c6d
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions deepmd/tf/descriptor/se_atten.py
Original file line number Diff line number Diff line change
Expand Up @@ -686,9 +686,7 @@ def _pass_filter(
tf.cast(mask, tf.bool),
inputs_i,
# broadcast is available: (nframes * nloc, 1) -> (nframes * nloc, ndescrpt)
tf.reshape(
self.avg_looked_up, [-1, 1]
),
tf.reshape(self.avg_looked_up, [-1, 1]),
)
else:
inputs_i *= mask

Check warning on line 692 in deepmd/tf/descriptor/se_atten.py

View check run for this annotation

Codecov / codecov/patch

deepmd/tf/descriptor/se_atten.py#L692

Added line #L692 was not covered by tests
Expand Down

0 comments on commit 3314c6d

Please sign in to comment.