[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
deepmodeling · Oct 16, 2024 · 5abdcde · 5abdcde
1 parent 12d862d
commit 5abdcde
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 10 deletions.
diff --git a/deepmd/pt/model/task/fitting.py b/deepmd/pt/model/task/fitting.py
@@ -19,9 +19,6 @@
     FittingNet,
     NetworkCollection,
 )
-from deepmd.pt.model.network.network import (
-    ResidualDeep,
-)
 from deepmd.pt.model.task.base_fitting import (
     BaseFitting,
 )
@@ -472,14 +469,10 @@ def _forward_common(
             device=descriptor.device,
         )  # jit assertion
         if self.mixed_types:
-            atom_property = (
-                self.filter_layers.networks[0](xx) + self.bias_atom_e[atype]
-            )
+            atom_property = self.filter_layers.networks[0](xx) + self.bias_atom_e[atype]
             if xx_zeros is not None:
                 atom_property -= self.filter_layers.networks[0](xx_zeros)
-            outs = (
-                outs + atom_property
-            )  # Shape is [nframes, natoms[0], net_dim_out]
+            outs = outs + atom_property  # Shape is [nframes, natoms[0], net_dim_out]
         else:
             for type_i, ll in enumerate(self.filter_layers.networks):
                 mask = (atype == type_i).unsqueeze(-1)

diff --git a/source/tests/pt/model/test_ener_fitting.py b/source/tests/pt/model/test_ener_fitting.py
@@ -10,7 +10,6 @@
     DescrptSeA,
 )
 from deepmd.pt.model.task.ener import (
-    EnergyFittingNet,
     InvarFitting,
 )
 from deepmd.pt.utils import (