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

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

deepmd/dpmodel/output_def.py

@@ -3,10 +3,6 @@
 from enum import (
     IntEnum,
 )
-from typing import (
-    Optional,
-    Union,
-)
 
 
 def check_shape(

deepmd/entrypoints/test.py

@@ -783,7 +783,13 @@ def test_property(
     assert isinstance(property_name, str)
     data.add(property_name, dp.task_dim, atomic=False, must=True, high_prec=True)
     if has_atom_property:
-        data.add(f"atom_{property_name}", dp.task_dim, atomic=True, must=False, high_prec=True)
+        data.add(
+            f"atom_{property_name}",
+            dp.task_dim,
+            atomic=True,
+            must=False,
+            high_prec=True,
+        )
 
     if dp.get_dim_fparam() > 0:
         data.add(

deepmd/pt/model/model/property_model.py

@@ -1,7 +1,6 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 from typing import (
     Optional,
-    Union,
 )
 
 import torch
@@ -63,8 +62,12 @@ def forward(
             do_atomic_virial=do_atomic_virial,
         )
         model_predict = {}
-        model_predict[f"atom_{self.get_property_name()}"] = model_ret[self.get_property_name()]
-        model_predict[self.get_property_name()] = model_ret[f"{self.get_property_name()}_redu"]
+        model_predict[f"atom_{self.get_property_name()}"] = model_ret[
+            self.get_property_name()
+        ]
+        model_predict[self.get_property_name()] = model_ret[
+            f"{self.get_property_name()}_redu"
+        ]
         if "mask" in model_ret:
             model_predict["mask"] = model_ret["mask"]
         return model_predict
@@ -108,8 +111,12 @@ def forward_lower(
             extra_nlist_sort=self.need_sorted_nlist_for_lower(),
         )
         model_predict = {}
-        model_predict[f"atom_{self.get_property_name()}"] = model_ret[self.get_property_name()]
-        model_predict[self.get_property_name()] = model_ret[f"{self.get_property_name()}_redu"]
+        model_predict[f"atom_{self.get_property_name()}"] = model_ret[
+            self.get_property_name()
+        ]
+        model_predict[self.get_property_name()] = model_ret[
+            f"{self.get_property_name()}_redu"
+        ]
         if "mask" in model_ret:
             model_predict["mask"] = model_ret["mask"]
         return model_predict

deepmd/pt/utils/stat.py

@@ -12,9 +12,6 @@
 import numpy as np
 import torch
 
-from deepmd.dpmodel.output_def import (
-    FittingOutputDef,
-)
 from deepmd.pt.utils import (
     AtomExcludeMask,
 )

deepmd/utils/out_stat.py

@@ -181,4 +181,4 @@ def compute_stats_property(
     output_std = output_std.reshape(var_shape)
     output_std = np.tile(output_std, (computed_output_bias.shape[0], 1))
 
-    return computed_output_bias, output_std
+    return computed_output_bias, output_std