[BUG] Fix ray wait in RayPartitionSet

[jaychia]

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.

[codspeed-hq]

CodSpeed Performance Report

Merging #3251 will degrade performances by 26.78%

_{Comparing jay/wait-no-local (a3ede70) with main (f566125)}

Summary

⚡ 1 improvements
❌ 2 regressions
✅ 14 untouched benchmarks

⚠️ Please fix the performance issues or acknowledge them on CodSpeed.

Benchmarks breakdown

	Benchmark	main	jay/wait-no-local	Change
❌	test_count[1 Small File]	3.2 ms	4.1 ms	-22.24%
⚡	test_iter_rows_first_row[100 Small Files]	358.8 ms	283.5 ms	+26.57%
❌	test_show[100 Small Files]	24 ms	32.8 ms	-26.78%

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 77.86%. Comparing base (f566125) to head (a3ede70).
Report is 41 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #3251   +/-   ##
=======================================
  Coverage   77.86%   77.86%           
=======================================
  Files         643      643           
  Lines       79558    79558           
=======================================
  Hits        61947    61947           
  Misses      17611    17611           

Files with missing lines	Coverage Δ
daft/runners/ray_runner.py	81.01% <100.00%> (ø)

... and 1 file with indirect coverage changes