14 Pull requests merged by 1 person

• Add checkpoint/_src/metadata/BUILD.

  #1471 merged Jan 13, 2025

• Internal

  #1467 merged Jan 9, 2025

• Fix typo in error message when unexpected best_mode passed in to checkpoint manager.

  #1482 merged Jan 9, 2025

• Add serialize_for_update() method for StepMetadata.

  #1474 merged Jan 9, 2025

• Eliminate usages of CompositeCheckpointHandler that use item_names or items_and_handlers.

  #1481 merged Jan 9, 2025

• Internal

  #1478 merged Jan 9, 2025

• Internal

  #1477 merged Jan 9, 2025

• Internal change.

  #1476 merged Jan 9, 2025

• Rework StepMetadata.item_metadata deserialization logic.

  #1472 merged Jan 8, 2025

• Add checkpoint/_src/path/BUILD

  #1466 merged Jan 8, 2025

• Remove item_metadata key serialization from StepMetadata.

  #1468 merged Jan 7, 2025

• Disable raise_array_data_missing_error in PyTreeCheckpointHandler restoration code path.

  #1469 merged Jan 7, 2025

• Internal

  #1465 merged Jan 7, 2025

• Fix a bug where a custom snapshot_dir is not passed to _release_snapshot, causing snapshots to not be released in this case.

  #1451 merged Jan 6, 2025

5 Pull requests opened by 1 person

• Internal change.

  #1479 opened Jan 9, 2025

• Internal change.

  #1483 opened Jan 10, 2025

• Internal

  #1485 opened Jan 13, 2025

• Add orbax/checkpoint/_src/BUILD

  #1486 opened Jan 13, 2025

• Fix dependencies on orbax/checkpoint/_src/multihost and orbax/checkpoint/_src/metadata as these two BUILDs are now added.

  #1487 opened Jan 13, 2025

2 Issues closed by 2 people

• orbax-checkpoint 0.10.3 breaks grain

  #1456 closed Jan 6, 2025

• AttributeError: module 'orbax.checkpoint.metadata' has no attribute 'Metadata'

  #1294 closed Jan 6, 2025

1 Issue opened by 1 person

• Checkpoint state invalid for both Zarr and OCDBT formats

  #1484 opened Jan 13, 2025

5 Unresolved conversations

Sometimes conversations happen on old items that aren’t yet closed. Here is a list of all the Issues and Pull Requests with unresolved conversations.

• Resubmit single array error message

  #1323 commented on Jan 10, 2025 • 0 new comments

• Internal

  #1406 commented on Jan 13, 2025 • 0 new comments

• Save `StepMetadata` in the `Checkpointer`.

  #1449 commented on Jan 11, 2025 • 0 new comments

• Support custom PyTree metadata.

  #1461 commented on Jan 8, 2025 • 0 new comments

• Add `_update_metadata()` method to `CheckpointHandler`s.

  #1463 commented on Jan 7, 2025 • 0 new comments