diff --git a/deepmd/main.py b/deepmd/main.py
index b2cdad5935..b8fa2cde39 100644
--- a/deepmd/main.py
+++ b/deepmd/main.py
@@ -765,7 +765,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