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

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

Author: pre-commit-ci[bot]

deepmd/__init__.py

@@ -15,7 +15,10 @@
 
 if TYPE_CHECKING:
     from deepmd.infer import DeepPotential as DeepPotentialType
-    from deepmd.property import PropertyPredictor, PropertyTrainer
+    from deepmd.property import (
+        PropertyPredictor,
+        PropertyTrainer,
+    )
 
 try:
     from deepmd._version import version as __version__
@@ -47,6 +50,7 @@ def DeepPotential(*args: Any, **kwargs: Any) -> "DeepPotentialType":
 
     return DeepPotential(*args, **kwargs)
 
+
 def __getattr__(name: str) -> Any:
     """Lazily expose optional high-level helpers.
 

deepmd/property.py

@@ -211,7 +211,7 @@ def from_systems(
         use_pretrain_script: bool | None = None,
         force_load: bool = False,
         **kwargs: Any,
-    ) -> "PropertyTrainer":
+    ) -> PropertyTrainer:
         trainer = cls(**kwargs)
         trainer._system_options = {
             "type_map": type_map,
@@ -324,9 +324,7 @@ def fit(self, data: Any | None = None) -> Path:
                 "Use PropertyTrainer.from_systems(...) first."
             )
 
-        from deepmd.pt.entrypoints.main import (
-            train as pt_train,
-        )
+        from deepmd.pt.entrypoints.main import train as pt_train
 
         config = self.build_input()
         finetune_model = resolve_model_name(

source/tests/common/test_property_helper.py

@@ -45,7 +45,9 @@ def test_build_input(self) -> None:
         self.assertEqual(config["model"]["type_map"], ["H", "C", "N", "O"])
         self.assertEqual(config["model"]["fitting_net"]["property_name"], "band_prop")
         self.assertEqual(config["model"]["fitting_net"]["task_dim"], 3)
-        self.assertEqual(config["training"]["training_data"]["systems"], ["train0", "train1"])
+        self.assertEqual(
+            config["training"]["training_data"]["systems"], ["train0", "train1"]
+        )
         self.assertEqual(config["training"]["validation_data"]["systems"], ["valid0"])
         self.assertEqual(config["training"]["validation_data"]["batch_size"], 4)