From bdbdc730d77c94948bec250f798b5544d1c66b8e Mon Sep 17 00:00:00 2001 From: Kevin Wang Date: Wed, 14 Feb 2024 14:03:07 -0800 Subject: [PATCH] add scheduler prints --- daft/runners/ray_runner.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/daft/runners/ray_runner.py b/daft/runners/ray_runner.py index d7a33ca66e..dc07f4ca5d 100644 --- a/daft/runners/ray_runner.py +++ b/daft/runners/ray_runner.py @@ -704,11 +704,13 @@ def __init__( max_task_backlog=max_task_backlog, use_ray_tqdm=True, ) + print("==== using remote scheduler ====") else: self.scheduler = Scheduler( max_task_backlog=max_task_backlog, use_ray_tqdm=False, ) + print("==== using local scheduler ====") print("==== Got to after scheduler creation ====")