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

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

deepmd/main.py

@@ -770,7 +770,9 @@ def main_parser() -> argparse.ArgumentParser:
         """
         ),
     )
-    parser_list_model_branch.add_argument("INPUT", help="The input multi-task pre-trained model file")
+    parser_list_model_branch.add_argument(
+        "INPUT", help="The input multi-task pre-trained model file"
+    )
 
     return parser
 
@@ -829,7 +831,7 @@ def main():
         "compress",
         "convert-from",
         "train-nvnmd",
-        "list-model-branch"
+        "list-model-branch",
     ):
         deepmd_main = BACKENDS[args.backend]().entry_point_hook
     elif args.command is None:

deepmd/pt/entrypoints/main.py

@@ -74,7 +74,6 @@
 from deepmd.utils.summary import SummaryPrinter as BaseSummaryPrinter
 
 log = logging.getLogger(__name__)
-from IPython import embed
 
 
 def get_trainer(
@@ -346,7 +345,7 @@ def main(args: Optional[Union[List[str], argparse.Namespace]] = None):
         #  Pretrained model must be multitask mode
         assert finetune_from_multi_task, "When using --list-model-branch, the pretrained model must be multitask model"
         model_branch = list(model_params["model_dict"].keys())
-        log.info(f"Available model branches are {model_branch}")        
+        log.info(f"Available model branches are {model_branch}")
     else:
         raise RuntimeError(f"Invalid command {FLAGS.command}!")
 

deepmd/pt/train/training.py

@@ -83,7 +83,9 @@
 )
 
 log = logging.getLogger(__name__)
-from IPython import embed
+from IPython import (
+    embed,
+)
 
 
 class Trainer:

deepmd/pt/utils/finetune.py

@@ -5,7 +5,6 @@
 )
 
 import torch
-from IPython import embed
 
 from deepmd.pt.utils import (
     env,