Better checkpoint metavar

ecmwf · Aug 1, 2024 · 69b76a6 · 69b76a6
1 parent c43ab56
commit 69b76a6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/anemoi/training/commands/checkpoint.py b/src/anemoi/training/commands/checkpoint.py
@@ -25,8 +25,8 @@ def add_arguments(self, command_parser: argparse.ArgumentParser):
             help=help,
             description=help,
         )
-        inference.add_argument("--input", "-i", required=True, metavar="input.ckpt")
-        inference.add_argument("--output", "-o", required=True, metavar="output.ckpt")
+        inference.add_argument("--input", "-i", required=True, metavar="training.ckpt")
+        inference.add_argument("--output", "-o", required=True, metavar="inference.ckpt")
 
     def run(self, args):
         if args.subcommand == "inference":