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

add compact_name_scope v3 #3646

Merged
merged 1 commit into from
Feb 5, 2024
Merged

add compact_name_scope v3 #3646

merged 1 commit into from
Feb 5, 2024

Conversation

cgarciae
Copy link
Collaborator

What does this PR do?

(Take 3 due to internal pytype bug)

Adds the nn.compact_name_scope decorator that enables methods to act as compact name scopes as with regular Haiku methods, this makes porting Haiku code easier.

@cgarciae cgarciae force-pushed the compact_name_scope_v3 branch 3 times, most recently from 6150a60 to 2980501 Compare January 25, 2024 12:45
@codecov-commenter
Copy link

Codecov Report

Attention: 21 lines in your changes are missing coverage. Please review.

Comparison is base (a93b7a8) 56.25% compared to head (2980501) 56.24%.

Files Patch % Lines
flax/linen/module.py 40.00% 21 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main    #3646      +/-   ##
==========================================
- Coverage   56.25%   56.24%   -0.01%     
==========================================
  Files         101      101              
  Lines       12184    12218      +34     
==========================================
+ Hits         6854     6872      +18     
- Misses       5330     5346      +16     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@chiamp
Copy link
Collaborator

chiamp commented Jan 26, 2024

There's a merge conflict. Can you rebase and then run TGP?

@cgarciae cgarciae force-pushed the compact_name_scope_v3 branch from 2980501 to b183dc6 Compare February 1, 2024 14:19
@copybara-service copybara-service bot merged commit 48f3a70 into main Feb 5, 2024
21 checks passed
@copybara-service copybara-service bot deleted the compact_name_scope_v3 branch February 5, 2024 18:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants