Skip to content

[AIR] Fix CountVectorizer failing with big data - #32351

Merged
amogkam merged 4 commits into
ray-project:masterfrom
Yard1:count_vectorizer_fix
Feb 16, 2023
Merged

[AIR] Fix CountVectorizer failing with big data#32351
amogkam merged 4 commits into
ray-project:masterfrom
Yard1:count_vectorizer_fix

Conversation

@Yard1

@Yard1 Yard1 commented Feb 9, 2023

Copy link
Copy Markdown
Member

Signed-off-by: Antoni Baum antoni.baum@protonmail.com

Why are these changes needed?

Fixes bad return type in an internal CountVectorizer logic that caused it to fail if a Dataset task processed multiple batches instead of just one. Test has been updated.

Related issue number

Closes https://discuss.ray.io/t/xgboost-ray-crashes-when-used-for-multiclass-text-classification/9248/3

Checks

  • 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 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 :(

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>

@bveeramani bveeramani left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

Comment thread python/ray/data/tests/preprocessors/test_vectorizer.py Outdated
Comment thread python/ray/data/tests/preprocessors/test_vectorizer.py
Comment thread python/ray/data/preprocessors/vectorizer.py
Yard1 and others added 2 commits February 9, 2023 21:06
Co-authored-by: Balaji Veeramani <bveeramani@berkeley.edu>
Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
@amogkam
amogkam merged commit 5161c37 into ray-project:master Feb 16, 2023
@Yard1
Yard1 deleted the count_vectorizer_fix branch February 16, 2023 19:29
edoakes pushed a commit to edoakes/ray that referenced this pull request Mar 22, 2023
Fixes bad return type in an internal CountVectorizer logic that caused it to fail if a Dataset task processed multiple batches instead of just one. Test has been updated.

---------

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
Co-authored-by: Balaji Veeramani <bveeramani@berkeley.edu>
Signed-off-by: Edward Oakes <ed.nmi.oakes@gmail.com>
peytondmurray pushed a commit to peytondmurray/ray that referenced this pull request Mar 22, 2023
Fixes bad return type in an internal CountVectorizer logic that caused it to fail if a Dataset task processed multiple batches instead of just one. Test has been updated.

---------

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
Co-authored-by: Balaji Veeramani <bveeramani@berkeley.edu>
elliottower pushed a commit to elliottower/ray that referenced this pull request Apr 22, 2023
Fixes bad return type in an internal CountVectorizer logic that caused it to fail if a Dataset task processed multiple batches instead of just one. Test has been updated.

---------

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
Co-authored-by: Balaji Veeramani <bveeramani@berkeley.edu>
Signed-off-by: elliottower <elliot@elliottower.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants