From e8448345f11f51878c68b5ed4063a7b5d11ae59a Mon Sep 17 00:00:00 2001
From: Peter Rowlands <peter@pmrowla.com>
Date: Thu, 31 Mar 2022 11:59:48 +0900
Subject: [PATCH] exp: follow output on run-all

---
 dvc/repo/experiments/__init__.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/dvc/repo/experiments/__init__.py b/dvc/repo/experiments/__init__.py
index 680bd6f1a..987ec9d0d 100644
--- a/dvc/repo/experiments/__init__.py
+++ b/dvc/repo/experiments/__init__.py
@@ -169,6 +169,12 @@ class Experiments:
 
     def reproduce_queued(self, **kwargs):
         self.celery_queue.spawn_worker()
+        # wait for task execution to start
+        while not first(self.celery_queue.iter_active()):
+            time.sleep(1)
+        for entry in self.celery_queue.iter_active():
+            for line in self.celery_queue.proc.follow(entry.stash_rev):
+                ui.write(line, end="")
 
     def _log_reproduced(self, revs: Iterable[str], tmp_dir: bool = False):
         names = []