Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[BUG] Fix ray wait in RayPartitionSet (#3251)
Closes: #3249 Ray's `ray.wait` is supposed to: 1. Defaults to `fetch_local=True` which will supposedly fetch data to wherever the wait is called before returning 2. Defaults to `num_returns=1` which will wait until only the first item is ready before returning This seems to not be the intended behavior here, where `RayPartitionSet` is trying to wait on ALL the partitions to be ready, and does not want to pull any data down to the calling site. Co-authored-by: Jay Chia <jaychia94@gmail.com@users.noreply.github.com>
- Loading branch information