Nothing here, but needed for diffbase

PiperOrigin-RevId: 713835965
google · Feb 3, 2025 · b6c4fe7 · b6c4fe7
1 parent af0ad20
commit b6c4fe7
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/checkpoint/orbax/checkpoint/_src/checkpointers/async_checkpointer.py b/checkpoint/orbax/checkpoint/_src/checkpointers/async_checkpointer.py
@@ -37,6 +37,7 @@
 from orbax.checkpoint._src.path import atomicity_types
 
 
+
 BarrierSyncFn = multihost.BarrierSyncFn
 _DIRECTORY_CREATION_SIGNALS = [
     synchronization.HandlerAwaitableSignal.STEP_DIRECTORY_CREATION
@@ -49,7 +50,8 @@ def _on_commit_callback(
 ):
   """Finalize atomic save and record checkpoint save metrics."""
   atomicity.on_commit_callback(
-      tmpdir, checkpoint_start_time=checkpoint_start_time
+      tmpdir,
+      checkpoint_start_time=checkpoint_start_time,
   )
   total_duration_secs = time.time() - checkpoint_start_time
   jax.monitoring.record_event_duration_secs(
@@ -441,7 +443,10 @@ def _callback() -> None:
           threading.current_thread().name,
           tmpdir.get(),
       )
-      _on_commit_callback(tmpdir, checkpoint_start_time)
+      _on_commit_callback(
+          tmpdir,
+          checkpoint_start_time,
+      )
 
     self._async_manager.start_async_commit(
         directory,

diff --git a/checkpoint/orbax/checkpoint/checkpoint_manager.py b/checkpoint/orbax/checkpoint/checkpoint_manager.py
@@ -56,6 +56,7 @@
 from typing_extensions import Self  # for Python version < 3.11
 
 
+
 PyTree = Any
 CheckpointDirs = Tuple[str, str]
 SaveParams = Mapping[str, Any]

diff --git a/checkpoint/orbax/checkpoint/options.py b/checkpoint/orbax/checkpoint/options.py
@@ -70,5 +70,3 @@ class FileOptions:
 
   path_permission_mode: Optional[int] = None
 
-
-
Original file line number	Diff line number	Diff line change
Expand Up		@@ -70,5 +70,3 @@ class FileOptions:

		path_permission_mode: Optional[int] = None