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

Remove string from checkpointing example and unbreak doctest in head #3288

Merged
merged 2 commits into from
Aug 22, 2023

Conversation

IvyZX
Copy link
Collaborator

@IvyZX IvyZX commented Aug 21, 2023

Orbax construct_restore_args doesn't work for strings. This should be fixed on the next release. Removing the string from checkpointing example for now.

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@codecov-commenter
Copy link

codecov-commenter commented Aug 21, 2023

Codecov Report

Merging #3288 (78a9727) into main (a767363) will increase coverage by 0.03%.
Report is 2 commits behind head on main.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##             main    #3288      +/-   ##
==========================================
+ Coverage   82.58%   82.61%   +0.03%     
==========================================
  Files          55       55              
  Lines        6315     6317       +2     
==========================================
+ Hits         5215     5219       +4     
+ Misses       1100     1098       -2     
Files Changed Coverage Δ
flax/linen/spmd.py 90.83% <ø> (ø)
flax/linen/module.py 92.13% <100.00%> (+0.27%) ⬆️

@IvyZX IvyZX requested review from cgarciae and chiamp August 21, 2023 23:37
@copybara-service copybara-service bot merged commit 9e82cdb into google:main Aug 22, 2023
16 checks passed
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