Add a presubmit benchmark for CheckpointManager using Pathways.

PiperOrigin-RevId: 825533145

Author: Orbax Authors

checkpoint/orbax/checkpoint/_src/testing/benchmarks/checkpoint_manager_benchmark.py

@@ -22,6 +22,7 @@
 import numpy as np
 from orbax.checkpoint import args as args_lib
 from orbax.checkpoint import checkpoint_manager
+from orbax.checkpoint import multihost
 from orbax.checkpoint import utils
 from orbax.checkpoint._src.testing.benchmarks.core import core as benchmarks_core
 from orbax.checkpoint._src.testing.benchmarks.core import metric as metric_lib
@@ -55,6 +56,7 @@ def test_fn(
     options = context.options
     assert isinstance(options, CheckpointManagerBenchmarkOptions)
 
+
     cm_options = checkpoint_manager.CheckpointManagerOptions(
         save_interval_steps=options.save_interval_steps,
         max_to_keep=options.max_to_keep,