Skip to content

[Data] Compute Expressions-fixed size array#58741

Merged
bveeramani merged 28 commits into
ray-project:masterfrom
400Ping:data/compute-expressions-arr
Jan 14, 2026
Merged

[Data] Compute Expressions-fixed size array#58741
bveeramani merged 28 commits into
ray-project:masterfrom
400Ping:data/compute-expressions-arr

Conversation

@400Ping

@400Ping 400Ping commented Nov 18, 2025

Copy link
Copy Markdown
Contributor

Description

Completing the fixed-size array namespace operations

Related issues

Related to #58674

Additional information

@400Ping 400Ping requested a review from a team as a code owner November 18, 2025 15:44

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Code Review

This pull request introduces a new _ArrayNamespace to handle fixed-size array operations in Ray Data expressions, adding flatten() and to_list() methods. The implementation is a good start, but I've identified a couple of issues in the new arr_namespace.py file. Specifically, the flatten() method has incorrect fallback logic that makes its behavior inconsistent, and the to_list() method can be optimized for better performance. My review includes suggestions to fix these issues for improved correctness and efficiency.

Comment thread python/ray/data/namespace_expressions/arr_namespace.py Outdated
Comment thread python/ray/data/namespace_expressions/arr_namespace.py
Comment thread python/ray/data/expressions.py Outdated
Comment thread python/ray/data/namespace_expressions/arr_namespace.py Outdated
@400Ping 400Ping marked this pull request as draft November 18, 2025 16:37
@400Ping 400Ping marked this pull request as ready for review November 18, 2025 16:38
@400Ping 400Ping force-pushed the data/compute-expressions-arr branch from a86b794 to 000cf1e Compare November 18, 2025 16:42
@400Ping 400Ping changed the title [Data]Compute Expressions-fixed size array Nov 18, 2025
@400Ping 400Ping changed the title [WIP[Data]Compute Expressions-fixed size array Nov 18, 2025
@ray-gardener ray-gardener Bot added data Ray Data-related issues community-contribution Contributed by the community labels Nov 18, 2025
@gvspraveen

Copy link
Copy Markdown
Contributor

Hi @400Ping Thanks for contribution. Can you fix the CI (lint issue)?

@iamjustinhsu iamjustinhsu left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Thanks @400Ping! Can you also add some tests?

Comment thread python/ray/data/namespace_expressions/arr_namespace.py Outdated
Comment thread python/ray/data/namespace_expressions/arr_namespace.py Outdated
Comment thread python/ray/data/expressions.py Outdated
Comment thread python/ray/data/namespace_expressions/arr_namespace.py Outdated
@400Ping 400Ping marked this pull request as draft November 20, 2025 10:27
Signed-off-by: 400Ping <fourhundredping@gmail.com>
Signed-off-by: 400Ping <fourhundredping@gmail.com>
Signed-off-by: 400Ping <fourhundredping@gmail.com>
Signed-off-by: 400Ping <fourhundredping@gmail.com>
@400Ping 400Ping force-pushed the data/compute-expressions-arr branch from 987ca07 to 7819f4b Compare November 21, 2025 18:38
Signed-off-by: 400Ping <fourhundredping@gmail.com>
@400Ping 400Ping changed the title [WIP][Data]Compute Expressions-fixed size array Nov 22, 2025
@400Ping 400Ping marked this pull request as ready for review November 22, 2025 05:56
Comment thread python/ray/data/namespace_expressions/arr_namespace.py Outdated
@400Ping

400Ping commented Nov 23, 2025

Copy link
Copy Markdown
Contributor Author
Signed-off-by: 400Ping <fourhundredping@gmail.com>
@400Ping 400Ping requested a review from iamjustinhsu November 25, 2025 06:03
Comment thread python/ray/data/namespace_expressions/arr_namespace.py Outdated
Signed-off-by: Ping <fourhundredping@gmail.com>
Signed-off-by: 400Ping <fourhundredping@gmail.com>
Comment thread python/ray/data/tests/test_namespace_expressions.py Outdated
@400Ping

400Ping commented Dec 7, 2025

Copy link
Copy Markdown
Contributor Author

@goutamvenkat-anyscale fixed, PTAL.

@400Ping

400Ping commented Dec 11, 2025

Copy link
Copy Markdown
Contributor Author
@400Ping

400Ping commented Dec 11, 2025

Copy link
Copy Markdown
Contributor Author
Comment thread python/ray/data/expressions.py Outdated
Signed-off-by: 400Ping <fourhundredping@gmail.com>
Comment thread python/ray/data/namespace_expressions/list_namespace.py Outdated
Signed-off-by: 400Ping <fourhundredping@gmail.com>
@400Ping

400Ping commented Dec 17, 2025

Copy link
Copy Markdown
Contributor Author
Comment thread python/ray/data/namespace_expressions/list_namespace.py Outdated
@400Ping

400Ping commented Dec 31, 2025

Copy link
Copy Markdown
Contributor Author
@400Ping

400Ping commented Jan 8, 2026

Copy link
Copy Markdown
Contributor Author
@owenowenisme

Copy link
Copy Markdown
Member

@400Ping Could you fix the conflict first? Thanks.

Signed-off-by: 400Ping <fourhundredping@gmail.com>
@400Ping

400Ping commented Jan 10, 2026

Copy link
Copy Markdown
Contributor Author

@400Ping Could you fix the conflict first? Thanks.

Done

@goutamvenkat-anyscale goutamvenkat-anyscale left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Overall looks good. Thanks.

Just a couple of comments

Comment thread python/ray/data/tests/expressions/test_namespace_list.py Outdated
Signed-off-by: 400Ping <fourhundredping@gmail.com>

@goutamvenkat-anyscale goutamvenkat-anyscale left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

lgtm

Signed-off-by: 400Ping <fourhundredping@gmail.com>
@bveeramani bveeramani merged commit 28bca7a into ray-project:master Jan 14, 2026
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

community-contribution Contributed by the community data Ray Data-related issues @external-author-action-required Alternate tag for PRs where the author doesn't have labeling permission. go add ONLY when ready to merge, run all tests

8 participants