Fix missing default

metatensor · Feb 1, 2024 · 853f1d7 · 853f1d7
1 parent 75c1408
commit 853f1d7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/metatensor/models/cli/eval_model.py b/src/metatensor/models/cli/eval_model.py
@@ -42,7 +42,7 @@ def _add_eval_model_parser(subparser: argparse._SubParsersAction) -> None:
     )
 
 
-def eval_model(model: str, structures: str, output: str = "output.xyz") -> None:
+def eval_model(model: str, structures: str, output: str) -> None:
     """Evaluate a pretrained model.
 
     ``target_property`` wil be predicted on a provided set of structures. Predicted

diff --git a/src/metatensor/models/cli/export_model.py b/src/metatensor/models/cli/export_model.py
@@ -30,11 +30,12 @@ def _add_export_model_parser(subparser: argparse._SubParsersAction) -> None:
         dest="output",
         type=str,
         required=False,
+        default = "exported-model.pt",
         help="Filename of the exported model (default: %(default)s).",
     )
 
 
-def export_model(model: str, output: str = "exported-model.pt") -> None:
+def export_model(model: str, output: str) -> None:
     """Export a pretrained model to run MD simulations
 
     :param model: Path to a saved model