Skip to content

Commit

Permalink
add debugging prints
Browse files Browse the repository at this point in the history
  • Loading branch information
kevinzwang committed Feb 14, 2024
1 parent 1066ace commit 3af08c8
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions daft/runners/ray_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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())
Expand Down

0 comments on commit 3af08c8

Please sign in to comment.