-
Notifications
You must be signed in to change notification settings - Fork 6.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Make sure all-to-all operator return num outputs so progress bar can work #31825
Conversation
Signed-off-by: jianoaix <iamjianxiao@gmail.com>
@@ -226,6 +227,14 @@ def _sleep(block_iter: Iterable[Block]) -> Iterable[Block]: | |||
wait_for_condition(lambda: (ray.available_resources().get("GPU", 0) == 1.0)) | |||
|
|||
|
|||
def test_prograss_bar(): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
why this unit test is testing progress bar? didn't see any assertion around progress bar, so asking.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It's a bit indirect because the progress bar is a local variable that's involved in execution. Make it test num outputs for operators.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
late LGTM, thanks @jianoaix.
Why are these changes needed?
To enable the new bulk execution backend: #30903
Based on the most recent test (https://buildkite.com/ray-project/oss-ci-build-pr/builds/9947#_), this should be last issue to fix it!
(note the failure of Dataset tests is not real as all tests passing, some issue with bazel test)
Related issue number
Checks
git commit -s
) in this PR.scripts/format.sh
to lint the changes in this PR.