diff --git a/src/exo/worker/tests/unittests/test_runner/test_runner_supervisor.py b/src/exo/worker/tests/unittests/test_runner/test_runner_supervisor.py index 30b8f833..c057bfe2 100644 --- a/src/exo/worker/tests/unittests/test_runner/test_runner_supervisor.py +++ b/src/exo/worker/tests/unittests/test_runner/test_runner_supervisor.py @@ -69,6 +69,7 @@ def _build_supervisor( _ev_send, ev_recv = mp_channel[Event]() task_sender, _task_recv = mp_channel[Task]() + cancel_sender, _cancel_recv = mp_channel[TaskId]() runner_process = multiprocessing.Process(target=target, daemon=True) heartbeat: Synchronized[int] = multiprocessing.Value("Q", 0) @@ -79,6 +80,7 @@ def _build_supervisor( initialize_timeout=10, _ev_recv=ev_recv, _task_sender=task_sender, + _cancel_sender=cancel_sender, _event_sender=event_sender.clone(), _heartbeat=heartbeat, )