#v1 Add Checkpointer.wait method.

PiperOrigin-RevId: 829067123

Author: cpgaffney1

checkpoint/CHANGELOG.md

@@ -7,8 +7,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.11.28] - 2025-11-06
+
 ### Added
 
+- #v1 `Checkpointer.wait` method.
 - Added `register_pathways_handlers` to `ocp.type_handlers`
 
 ## [0.11.27] - 2025-11-04

checkpoint/orbax/checkpoint/experimental/v1/_src/training/checkpointer.py

@@ -518,6 +518,10 @@ def is_saving_in_progress(self) -> bool:
     """Whether a checkpoint is currently being saved in the background."""
     return self._manager.is_saving_in_progress()
 
+  def wait(self):
+    """Waits for any outstanding async operations to complete."""
+    self._manager.wait_until_finished()
+
   def close(self):
     """Ensures any outstanding async operations are completed before closing."""
     self._manager.close()

checkpoint/orbax/checkpoint/version.py

@@ -17,7 +17,7 @@
 # A new PyPI release will be pushed everytime `__version__` is increased.
 # Also modify version and date in CHANGELOG.
 # LINT.IfChange
-__version__ = '0.11.27'
+__version__ = '0.11.28'
 # LINT.ThenChange(//depot//orbax/checkpoint/CHANGELOG.md)