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 12, 2024
1 parent 3021f6b commit 7893fa2
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions deepmd/descriptor/se_a_mask.py
Original file line number Diff line number Diff line change
Expand Up @@ -536,7 +536,7 @@ def _filter_lower(
is_exclude=False,
):
"""Input env matrix, returns R.G."""
outputs_size = [1] + self.filter_neuron
outputs_size = [1, *self.filter_neuron]
# cut-out inputs
# with natom x (nei_type_i x 4)
inputs_i = paddle.slice(
Expand Down Expand Up @@ -663,13 +663,11 @@ def _filter(
nframes = 1
# natom x (nei x 4)
shape = inputs.shape
outputs_size = [1] + self.filter_neuron
outputs_size = [1, *self.filter_neuron]
outputs_size_2 = self.n_axis_neuron # 16
all_excluded = all(
[
(type_input, type_i) in self.exclude_types # set()
(type_input, type_i) in self.exclude_types # set()
for type_i in range(self.ntypes)
]
) # False
if all_excluded:
# all types are excluded so result and qmat should be zeros
Expand Down

0 comments on commit 7893fa2

Please sign in to comment.