hotfix: added choices to argparse

charnley · Nov 5, 2023 · d192334 · d192334
1 parent dcca78e
commit d192334
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rmsd/calculate_rmsd.py b/rmsd/calculate_rmsd.py
@@ -1724,6 +1724,7 @@ def parse_arguments(arguments: Optional[Union[str, List[str]]] = None) -> argpar
             "Default is Kabsch."
         ),
         metavar="METHOD",
+        choices=ROTATION_METHODS,
     )
 
     # Reorder arguments
@@ -1743,6 +1744,7 @@ def parse_arguments(arguments: Optional[Union[str, List[str]]] = None) -> argpar
             f"{valid_reorder_methods}. "
             "Default is Hungarian."
         ),
+        choices=REORDER_METHODS,
     )
     parser.add_argument(
         "-ur",