[data] fix array contain nan element in group by #49420

Jay-ju · 2024-12-24T03:12:12Z

Why are these changes needed?

I've signed off every commit(by using the -s flag, i.e., git commit -s) in this PR.
I've run scripts/format.sh to lint the changes in this PR.
I've included any doc changes needed for https://docs.ray.io/en/master/.
- I've added any new APIs to the API Reference. For example, if I added a
  method in Tune, I've added it in doc/source/tune/api/ under the
  corresponding .rst file.
I've made sure the tests are passing. Note that there might be a few flaky tests, see the recent failures at https://flakey-tests.ray.io/
Testing Strategy
- Unit tests
- Release tests
- This PR is not tested :(

python/ray/data/_internal/arrow_block.py

Signed-off-by: jukejian <jukejian@bytedance.com>

Jay-ju requested a review from a team as a code owner December 24, 2024 03:12

Jay-ju force-pushed the fix_nan_group branch 2 times, most recently from 729b983 to 5677b37 Compare December 24, 2024 06:44

richardliaw reviewed Dec 24, 2024

View reviewed changes

python/ray/data/_internal/arrow_block.py Outdated Show resolved Hide resolved

[data] fix array contain nan element in group by

84cee3a

Signed-off-by: jukejian <jukejian@bytedance.com>

Jay-ju force-pushed the fix_nan_group branch from 1e7c3d3 to 84cee3a Compare December 25, 2024 02:47