MilesCranmer commited on
Commit
98fa83e
1 Parent(s): 0d92619

Fix click API usage

Browse files
Files changed (1) hide show
  1. pysr/_cli/main.py +5 -1
pysr/_cli/main.py CHANGED
@@ -52,8 +52,12 @@ TEST_OPTIONS = {"main", "env", "jax", "torch", "cli"}
52
 
53
 
54
  @pysr.command("test", help="Run PySR test suite.")
55
- @click.argument("tests", nargs=-1, help="Choose from " + ", ".join(TEST_OPTIONS) + ".")
56
  def _tests(tests):
 
 
 
 
57
  if len(tests) == 0:
58
  raise click.UsageError(
59
  "At least one test must be specified. "
 
52
 
53
 
54
  @pysr.command("test", help="Run PySR test suite.")
55
+ @click.argument("tests", nargs=-1)
56
  def _tests(tests):
57
+ """Run part of the PySR test suite.
58
+
59
+ Choose from main, env, jax, torch, and cli.
60
+ """
61
  if len(tests) == 0:
62
  raise click.UsageError(
63
  "At least one test must be specified. "