Skip to content

Commit 5abdcde

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

2 files changed

Lines changed: 2 additions & 10 deletions

File tree

deepmd/pt/model/task/fitting.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,6 @@
1919
FittingNet,
2020
NetworkCollection,
2121
)
22-
from deepmd.pt.model.network.network import (
23-
ResidualDeep,
24-
)
2522
from deepmd.pt.model.task.base_fitting import (
2623
BaseFitting,
2724
)
@@ -472,14 +469,10 @@ def _forward_common(
472469
device=descriptor.device,
473470
) # jit assertion
474471
if self.mixed_types:
475-
atom_property = (
476-
self.filter_layers.networks[0](xx) + self.bias_atom_e[atype]
477-
)
472+
atom_property = self.filter_layers.networks[0](xx) + self.bias_atom_e[atype]
478473
if xx_zeros is not None:
479474
atom_property -= self.filter_layers.networks[0](xx_zeros)
480-
outs = (
481-
outs + atom_property
482-
) # Shape is [nframes, natoms[0], net_dim_out]
475+
outs = outs + atom_property # Shape is [nframes, natoms[0], net_dim_out]
483476
else:
484477
for type_i, ll in enumerate(self.filter_layers.networks):
485478
mask = (atype == type_i).unsqueeze(-1)

source/tests/pt/model/test_ener_fitting.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
DescrptSeA,
1111
)
1212
from deepmd.pt.model.task.ener import (
13-
EnergyFittingNet,
1413
InvarFitting,
1514
)
1615
from deepmd.pt.utils import (

0 commit comments

Comments
 (0)