Skip to content

Commit

Permalink
Internal change.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 675577316
  • Loading branch information
niketkumar authored and Orbax Authors committed Sep 17, 2024
1 parent 7cf465e commit 1f62e57
Showing 1 changed file with 2 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -253,6 +253,8 @@ def _maybe_set_default_save_args(_, leaf_args):
)




class BasePyTreeCheckpointHandler(
async_checkpoint_handler.AsyncCheckpointHandler
):
Expand Down

0 comments on commit 1f62e57

Please sign in to comment.