From 3af08c853c2a358d3d021b8fe9f6afc48654fad6 Mon Sep 17 00:00:00 2001 From: Kevin Wang Date: Wed, 14 Feb 2024 10:21:09 -0800 Subject: [PATCH] add debugging prints --- daft/runners/ray_runner.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/daft/runners/ray_runner.py b/daft/runners/ray_runner.py index d765a313d5..691b8b5adc 100644 --- a/daft/runners/ray_runner.py +++ b/daft/runners/ray_runner.py @@ -696,6 +696,8 @@ def __init__( logger.warning(f"Ray has already been initialized, Daft will reuse the existing Ray context.") self.ray_context = ray.init(address=address, ignore_reinit_error=True) + print("==== Got to before scheduler creation ====") + if isinstance(self.ray_context, ray.client_builder.ClientContext): # Run scheduler remotely if the cluster is connected remotely. self.scheduler_actor = SchedulerActor.remote( # type: ignore @@ -708,6 +710,8 @@ def __init__( use_ray_tqdm=False, ) + print("==== Got to after scheduler creation ====") + def active_plans(self) -> list[str]: if isinstance(self.ray_context, ray.client_builder.ClientContext): return ray.get(self.scheduler_actor.active_plans.remote())