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

Make masking compatible with callable pytrees a la Equinox #1015

Merged
merged 1 commit into from
Jul 18, 2024

Conversation

copybara-service[bot]
Copy link

@copybara-service copybara-service bot commented Jul 17, 2024

Make masking compatible with callable pytrees a la Equinox

@copybara-service copybara-service bot force-pushed the test_653116911 branch 2 times, most recently from 7d315cf to c323822 Compare July 17, 2024 15:35
@copybara-service copybara-service bot changed the title Fix masking incompatibility with Equinox Make masking compatible with callable pytrees a la Equinox Jul 17, 2024
Copy link

@JadM133 JadM133 left a comment

Choose a reason for hiding this comment

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

Shouldn't we change the callable in "init_f" as well?

@copybara-service copybara-service bot force-pushed the test_653116911 branch 2 times, most recently from 0895649 to 4b93a26 Compare July 18, 2024 16:15
@copybara-service copybara-service bot merged commit 6f12911 into main Jul 18, 2024
@copybara-service copybara-service bot deleted the test_653116911 branch July 18, 2024 16:30
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.

2 participants