exp: add queue-worker plumbing command

This commit is contained in:
Peter Rowlands
2022-07-12 21:39:49 +09:00
committed by Peter Rowlands (변기호)
parent a0ae9d3f45
commit cffd587952
2 changed files with 27 additions and 0 deletions
+2
View File
@@ -15,6 +15,7 @@ from dvc.commands.experiments import (
ls,
pull,
push,
queue_worker,
remove,
run,
show,
@@ -30,6 +31,7 @@ SUB_COMMANDS = [
ls,
pull,
push,
queue_worker,
remove,
run,
show,
+25
View File
@@ -0,0 +1,25 @@
import logging
from dvc.cli.command import CmdBase
logger = logging.getLogger(__name__)
class CmdQueueWorker(CmdBase):
"""Run the exp queue worker."""
def run(self):
self.repo.experiments.celery_queue.worker.start(self.args.name)
return 0
def add_parser(experiments_subparsers, parent_parser):
QUEUE_WORKER_HELP = "Run the exp queue worker."
parser = experiments_subparsers.add_parser(
"queue-worker",
parents=[parent_parser],
description=QUEUE_WORKER_HELP,
add_help=False,
)
parser.add_argument("name", help="Celery worker name.")
parser.set_defaults(func=CmdQueueWorker)