FreakU
/
takahe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add run-for option to console runstator

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, default=30,
help="How often to run cleaning and scheduling", 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) parser.add_argument("model_labels", nargs="*", type=str)
def handle( def handle(
@ -41,6 +48,7 @@ class Command(BaseCommand):
concurrency: int, concurrency: int,
liveness_file: str, liveness_file: str,
schedule_interval: int, schedule_interval: int,
run_for: int,
*args, *args,
**options **options
): ):
@ -60,5 +68,6 @@ class Command(BaseCommand):
concurrency=concurrency, concurrency=concurrency,
liveness_file=liveness_file, liveness_file=liveness_file,
schedule_interval=schedule_interval, schedule_interval=schedule_interval,
run_for=run_for,
) )
async_to_sync(runner.run)() async_to_sync(runner.run)()