From cffd58795294eb651ce9f401f49b2fae10964263 Mon Sep 17 00:00:00 2001
From: Peter Rowlands <peter@pmrowla.com>
Date: Wed, 23 Feb 2022 16:11:08 +0900
Subject: [PATCH] exp: add queue-worker plumbing command

---
 dvc/commands/experiments/__init__.py     |  2 ++
 dvc/commands/experiments/queue_worker.py | 25 ++++++++++++++++++++++++
 2 files changed, 27 insertions(+)
 create mode 100644 dvc/commands/experiments/queue_worker.py

diff --git a/dvc/commands/experiments/__init__.py b/dvc/commands/experiments/__init__.py
index 633310942..76fed59ec 100644
--- a/dvc/commands/experiments/__init__.py
+++ b/dvc/commands/experiments/__init__.py
@@ -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,
diff --git a/dvc/commands/experiments/queue_worker.py b/dvc/commands/experiments/queue_worker.py
new file mode 100644
index 000000000..81b6372d9
--- /dev/null
+++ b/dvc/commands/experiments/queue_worker.py
@@ -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)