mirrors/takahe
1
1
Fork 0
mirror of https://github.com/jointakahe/takahe.git synced 2024-11-21 23:01:00 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add run-for option to console runstator

Browse source
This commit is contained in:
Andrew Godwin 2022-11-27 11:03:52 -07:00
parent 18c85bc6d2
commit 263af996d8
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
stator/management/commands/runstator.py
View file

@ -33,6 +33,13 @@ class Command(BaseCommand):
default=30,
help="How often to run cleaning and scheduling",
)
parser.add_argument(
"--run-for",
"-r",
type=int,
default=0,
help="How long to run for before exiting (defaults to infinite)",
)
parser.add_argument("model_labels", nargs="*", type=str)
def handle(
@ -41,6 +48,7 @@ class Command(BaseCommand):
concurrency: int,
liveness_file: str,
schedule_interval: int,
run_for: int,
*args,
**options
):
@ -60,5 +68,6 @@ class Command(BaseCommand):
concurrency=concurrency,
liveness_file=liveness_file,
schedule_interval=schedule_interval,
run_for=run_for,
)
async_to_sync(runner.run)()