Spaces:
Sleeping
Sleeping
Fix click API usage
Browse files- 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
|
| 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. "
|