From 6cfd1480a79981e55825a5e07c66c613db7ec5a7 Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Tue, 10 Jan 2023 12:26:24 +0800
Subject: [PATCH] scheduler: support passing event loop

---
 artiq/master/scheduler.py | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/artiq/master/scheduler.py b/artiq/master/scheduler.py
index 0d007954a..d849880fe 100644
--- a/artiq/master/scheduler.py
+++ b/artiq/master/scheduler.py
@@ -332,10 +332,10 @@ class Pipeline:
         self._run = RunStage(self.pool, deleter.delete)
         self._analyze = AnalyzeStage(self.pool, deleter.delete)
 
-    def start(self):
-        self._prepare.start()
-        self._run.start()
-        self._analyze.start()
+    def start(self, loop=None):
+        self._prepare.start(loop)
+        self._run.start(loop)
+        self._analyze.start(loop)
 
     async def stop(self):
         # NB: restart of a stopped pipeline is not supported
@@ -410,8 +410,9 @@ class Scheduler:
         self._deleter = Deleter(self._pipelines)
         self._log_submissions = log_submissions
 
-    def start(self):
-        self._deleter.start()
+    def start(self, loop=None):
+        self._loop = loop
+        self._deleter.start(self._loop)
 
     async def stop(self):
         # NB: restart of a stopped scheduler is not supported
@@ -442,7 +443,7 @@ class Scheduler:
                                 self._worker_handlers, self.notifier,
                                 self._experiment_db, self._log_submissions)
             self._pipelines[pipeline_name] = pipeline
-            pipeline.start()
+            pipeline.start(self._loop)
         return pipeline.pool.submit(expid, priority, due_date, flush, pipeline_name)
 
     def delete(self, rid):