Revert "[Data] Add configurable batching for resolve_block_refs to sp…#60114
Conversation
…eed up iter_batches (ray-project#58467)" This reverts commit 2a042d4. Signed-off-by: iamjustinhsu <jhsu@anyscale.com>
7d78ccd to
bd662d6
Compare
There was a problem hiding this comment.
Code Review
This pull request reverts a previous change that introduced configurable batching for resolve_block_refs. The revert appears to be clean and complete, correctly removing the feature logic, configuration, and associated tests. I have one minor suggestion to improve code clarity and avoid redundant calls in iter_batches.py.
| self._eager_free = ( | ||
| clear_block_after_read and DataContext.get_current().eager_free | ||
| ) | ||
|
|
||
| actor_prefetcher_enabled = ( | ||
| prefetch_batches > 0 | ||
| and self._ctx.actor_prefetcher_enabled | ||
| and DataContext.get_current().actor_prefetcher_enabled | ||
| and not ray.util.client.ray.is_connected() | ||
| ) |
There was a problem hiding this comment.
To avoid multiple calls to DataContext.get_current(), you could store the context in a local variable at the beginning of the method and reuse it. This improves readability and avoids repeated function calls.
ctx = DataContext.get_current()
self._eager_free = (
clear_block_after_read and ctx.eager_free
)
actor_prefetcher_enabled = (
prefetch_batches > 0
and ctx.actor_prefetcher_enabled
and not ray.util.client.ray.is_connected()
)|
Hey @YoussefEssDS, we had to revert this PR because the hurts performance when using
If you have any questions, feel free to reach out. You can also join the external ray data slack channel and we chat over there too |
ray-project#60114) …eed up iter_batches (ray-project#58467)" This reverts commit 2a042d4. ## Description Reverts # 58467 ## Related issues ## Additional information Signed-off-by: iamjustinhsu <jhsu@anyscale.com>
ray-project#60114) …eed up iter_batches (ray-project#58467)" This reverts commit 2a042d4. ## Description Reverts # 58467 ## Related issues ## Additional information Signed-off-by: iamjustinhsu <jhsu@anyscale.com> Signed-off-by: jeffery4011 <jefferyshen1015@gmail.com>
ray-project#60114) …eed up iter_batches (ray-project#58467)" This reverts commit 2a042d4. ## Description Reverts # 58467 ## Related issues ## Additional information Signed-off-by: iamjustinhsu <jhsu@anyscale.com>
ray-project#60114) …eed up iter_batches (ray-project#58467)" This reverts commit 2a042d4. ## Description Reverts # 58467 ## Related issues ## Additional information Signed-off-by: iamjustinhsu <jhsu@anyscale.com> Signed-off-by: peterxcli <peterxcli@gmail.com>
ray-project#60114) …eed up iter_batches (ray-project#58467)" This reverts commit 2a042d4. ## Description Reverts # 58467 ## Related issues ## Additional information Signed-off-by: iamjustinhsu <jhsu@anyscale.com> Signed-off-by: peterxcli <peterxcli@gmail.com>
…eed up iter_batches (#58467)"
This reverts commit 2a042d4.
Description
Reverts # 58467
Related issues
Additional information