Skip to content
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

[data] refactor StageSummaryStats for stage deprecation #41873

Merged
merged 4 commits into from
Dec 21, 2023

Conversation

Zandew
Copy link
Contributor

@Zandew Zandew commented Dec 13, 2023

Why are these changes needed?

Updates execution summary string to be N tasks executed, N blocks produced. The "total" blocks was removed since it was incorrect.

Fix tasks per node. Previously we were assuming tasks produce one block, so we were outputting blocks per node.

Adds rows per task to the output

Renames StageStatsSummary to OperatorStatsSummary.

Related issue number

Closes #41280

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

Signed-off-by: Andrew Xue <andewzxue@gmail.com>
Signed-off-by: Andrew Xue <andewzxue@gmail.com>
Signed-off-by: Andrew Xue <andewzxue@gmail.com>
Comment on lines -906 to -914
if len(exec_stats) < len(block_metas):
if exec_stats:
exec_summary_str += ", "
num_inherited = len(block_metas) - len(exec_stats)
exec_summary_str += "{}/{} blocks split from parent".format(
num_inherited, len(block_metas)
)
if not exec_stats:
exec_summary_str += " in {}s".format(rounded_total)
Copy link
Contributor Author

Choose a reason for hiding this comment

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

This should have been removed in #26363, this code path is never used.

python/ray/data/_internal/stats.py Show resolved Hide resolved
python/ray/data/_internal/stats.py Show resolved Hide resolved
python/ray/data/_internal/stats.py Outdated Show resolved Hide resolved
python/ray/data/_internal/stats.py Show resolved Hide resolved
Signed-off-by: Andrew Xue <andewzxue@gmail.com>
@Zandew Zandew marked this pull request as ready for review December 18, 2023 17:16
Copy link
Contributor

@stephanie-wang stephanie-wang left a comment

Choose a reason for hiding this comment

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

Sorry for the late review. Thanks for this change, much needed!

@stephanie-wang stephanie-wang merged commit 124cba6 into ray-project:master Dec 21, 2023
14 of 15 checks passed
stephanie-wang pushed a commit that referenced this pull request Jan 2, 2024
Followup to #41873, update some remaining docstrings and failing tests which check the raw string output of OperatorStatsSummary.

Signed-off-by: Scott Lee <sjl@anyscale.com>
vickytsang pushed a commit to ROCm/ray that referenced this pull request Jan 12, 2024
…t#41873)

Updates execution summary string to be N tasks executed, N blocks produced. The "total" blocks was removed since it was incorrect.

Fix tasks per node. Previously we were assuming tasks produce one block, so we were outputting blocks per node.

Adds rows per task to the output

Renames StageStatsSummary to OperatorStatsSummary.
Related issue number

Closes ray-project#41280

---------

Signed-off-by: Andrew Xue <andewzxue@gmail.com>
vickytsang pushed a commit to ROCm/ray that referenced this pull request Jan 12, 2024
…oject#42147

Followup to ray-project#41873, update some remaining docstrings and failing tests which check the raw string output of OperatorStatsSummary.

Signed-off-by: Scott Lee <sjl@anyscale.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[data] Dataset.stats() is misleading when tasks produce multiple blocks
3 participants