diff --git a/drivers/py/driver.py b/drivers/py/driver.py index 3567a0362..b199d6658 100755 --- a/drivers/py/driver.py +++ b/drivers/py/driver.py @@ -201,8 +201,8 @@ def run_driver( "--mode", type=str, default="dummy", + choices=list(__drivers__.keys()), help="""Type of potential to be used to compute the potential and its derivatives. - Currently implemented: [dummy, harmonic] """, ) parser.add_argument(