From 0af74f34525502efebccedc46ea78a3800e13d93 Mon Sep 17 00:00:00 2001 From: Jay Chia Date: Mon, 28 Oct 2024 19:10:32 -0700 Subject: [PATCH] Remove task not ready traces which are cluttering the file --- daft/runners/ray_runner.py | 5 +---- daft/runners/ray_tracing.py | 12 ------------ 2 files changed, 1 insertion(+), 16 deletions(-) diff --git a/daft/runners/ray_runner.py b/daft/runners/ray_runner.py index e44d8e667a..aeda1d5de0 100644 --- a/daft/runners/ray_runner.py +++ b/daft/runners/ray_runner.py @@ -796,7 +796,7 @@ def _await_tasks( timeout = 0.01 num_returns = len(inflight_ref_to_task_id) with runner_tracer.awaiting(num_returns, timeout): - readies, not_readies = ray.wait( + readies, _ = ray.wait( list(inflight_ref_to_task_id.keys()), num_returns=num_returns, timeout=timeout, @@ -808,9 +808,6 @@ def _await_tasks( if ready in inflight_ref_to_task_id: task_id = inflight_ref_to_task_id[ready] runner_tracer.task_received_as_ready(task_id, inflight_tasks[task_id].stage_id) - for not_ready in not_readies: - if not_ready in inflight_ref_to_task_id: - runner_tracer.task_not_ready(inflight_ref_to_task_id[not_ready]) return readies diff --git a/daft/runners/ray_tracing.py b/daft/runners/ray_tracing.py index d474f85440..b5361fd616 100644 --- a/daft/runners/ray_tracing.py +++ b/daft/runners/ray_tracing.py @@ -503,18 +503,6 @@ def task_dispatched(self, task_id: str): } ) - def task_not_ready(self, task_id: str): - self._write_event( - { - "id": task_id, - "category": "task", - "name": "task_awaited_not_ready", - "ph": PHASE_ASYNC_INSTANT, - "pid": 1, - "tid": 1, - } - ) - def task_received_as_ready(self, task_id: str, stage_id: int): self._write_event( {