hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix equalized split handling of num_splits == num_blocks case. (#23191)

Browse source
This commit is contained in:
Clark Zinzow 2022-03-15 12:23:50 -07:00 • committed by GitHub
parent 72713e815b
commit 1d5f18fe0a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/ray/data/dataset.py
View file

@ -632,7 +632,7 @@ class Dataset(Generic[T]):
This assume that the given splits are sorted in ascending order.
"""
if target_size == 0:
return splits
return splits, []
new_splits = []
leftovers = []
for split in splits: