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

for more information, see https://pre-commit.ci

deepmd/dpmodel/atomic_model/dp_atomic_model.py

@@ -25,6 +25,7 @@
     BaseAtomicModel,
 )
 
+
 @BaseAtomicModel.register("standard")
 class DPAtomicModel(BaseAtomicModel):
     """Model give atomic prediction of some physical property.

deepmd/dpmodel/atomic_model/linear_atomic_model.py

@@ -1,9 +1,5 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 import copy
-import sys
-from abc import (
-    abstractmethod,
-)
 from typing import (
     Dict,
     List,
@@ -353,9 +349,7 @@ def deserialize(cls, data) -> "DPZBLLinearEnergyAtomicModel":
         sw_rmin = data.pop("sw_rmin")
         sw_rmax = data.pop("sw_rmax")
         smin_alpha = data.pop("smin_alpha")
-        linear_model = LinearEnergyAtomicModel.deserialize(
-            data.pop("models")
-        )
+        linear_model = LinearEnergyAtomicModel.deserialize(data.pop("models"))
         dp_model, zbl_model = linear_model.models
         type_map = linear_model.type_map
 

deepmd/dpmodel/atomic_model/pairtab_atomic_model.py

@@ -24,6 +24,7 @@
     BaseAtomicModel,
 )
 
+
 @BaseAtomicModel.register("pairtab")
 class PairTabAtomicModel(BaseAtomicModel):
     """Pairwise tabulation energy model.

deepmd/pt/model/atomic_model/dp_atomic_model.py

@@ -32,6 +32,7 @@
 
 log = logging.getLogger(__name__)
 
+
 @BaseAtomicModel.register("standard")
 class DPAtomicModel(torch.nn.Module, BaseAtomicModel):
     """Model give atomic prediction of some physical property.

deepmd/pt/model/atomic_model/linear_atomic_model.py

@@ -1,9 +1,5 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 import copy
-import sys
-from abc import (
-    abstractmethod,
-)
 from typing import (
     Dict,
     List,
@@ -420,9 +416,7 @@ def deserialize(cls, data) -> "DPZBLLinearEnergyAtomicModel":
         sw_rmin = data.pop("sw_rmin")
         sw_rmax = data.pop("sw_rmax")
         smin_alpha = data.pop("smin_alpha")
-        linear_model = LinearEnergyAtomicModel.deserialize(
-            data.pop("models")
-        )
+        linear_model = LinearEnergyAtomicModel.deserialize(data.pop("models"))
         dp_model, zbl_model = linear_model.models
         type_map = linear_model.type_map
 

deepmd/pt/model/atomic_model/pairtab_atomic_model.py

@@ -34,6 +34,7 @@
     BaseAtomicModel,
 )
 
+
 @BaseAtomicModel.register("pairtab")
 class PairTabAtomicModel(torch.nn.Module, BaseAtomicModel):
     """Pairwise tabulation energy model.