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

for more information, see https://pre-commit.ci
deepmodeling · Mar 15, 2024 · 33e30ca · 33e30ca
1 parent 1fa3e18
commit 33e30ca
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/deepmd/pt/model/task/ener.py b/deepmd/pt/model/task/ener.py
@@ -22,15 +22,15 @@
     Fitting,
     GeneralFitting,
 )
+from deepmd.pt.model.task.invar_fitting import (
+    InvarFitting,
+)
 from deepmd.pt.utils import (
     env,
 )
 from deepmd.pt.utils.env import (
     DEFAULT_PRECISION,
 )
-from deepmd.pt.model.task.invar_fitting import(
-    InvarFitting
-)
 from deepmd.utils.version import (
     check_version_compatibility,
 )
@@ -41,7 +41,6 @@
 log = logging.getLogger(__name__)
 
 
-
 @Fitting.register("ener")
 class EnergyFittingNet(InvarFitting):
     def __init__(

diff --git a/deepmd/pt/model/task/invar_fitting.py b/deepmd/pt/model/task/invar_fitting.py
@@ -5,19 +5,16 @@
     Callable,
     List,
     Optional,
-    Tuple,
     Union,
 )
 
-import numpy as np
 import torch
 
 from deepmd.dpmodel import (
     FittingOutputDef,
     OutputVariableDef,
     fitting_check_output,
 )
-
 from deepmd.pt.model.task.fitting import (
     GeneralFitting,
 )