Skip to content

Fix handling of pytrees within jnp.array and jnp.asarray. #33042

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.

Sign up for GitHub

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

Jump to bottom
Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Fix handling of pytrees within jnp.array and jnp.asarray. #33042

wants to merge 1 commit into from
+31 −7

Conversation

@copybara-service
Copy link

@copybara-service copybara-service bot commented Oct 31, 2025

Fix handling of pytrees within jnp.array and jnp.asarray.

Previously, the logic meant for probing nested lists and tuples would traverse into the leaves of custom pytrees; this was unintentional. This change should make the handling of pytrees within jnp.array much more predictable.

Addresses #32725

@copybara-service copybara-service bot force-pushed the test_826571655 branch 4 times, most recently from 534bc68 to fffed9a Compare November 3, 2025 20:03
@Google-ML-Automation
Fix handling of pytrees within jnp.array and jnp.asarray.
a854e1a 
Previously, the logic meant for probing nested lists and tuples would traverse into the leaves of custom pytrees; this was unintentional. This change should make the handling of pytrees within `jnp.array` much more predictable.

Addresses #32725

PiperOrigin-RevId: 826571655
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

@jakevdp jakevdp

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@jakevdp