From 4d666dd6a2ba1844dea538d434c5ac2609c667ea Mon Sep 17 00:00:00 2001 From: Matej Aleksandrov Date: Tue, 30 Jun 2026 08:00:55 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 940456484 --- .../benchmarks/p2p_checkpoint_manager_benchmark_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/checkpoint/orbax/checkpoint/_src/testing/benchmarks/p2p_checkpoint_manager_benchmark_test.py b/checkpoint/orbax/checkpoint/_src/testing/benchmarks/p2p_checkpoint_manager_benchmark_test.py index d13cb28f2..6000a393c 100644 --- a/checkpoint/orbax/checkpoint/_src/testing/benchmarks/p2p_checkpoint_manager_benchmark_test.py +++ b/checkpoint/orbax/checkpoint/_src/testing/benchmarks/p2p_checkpoint_manager_benchmark_test.py @@ -317,12 +317,12 @@ def test_test_fn_applies_benchmark_options_correctly(self, options): class HelperFunctionsTest(parameterized.TestCase): - @mock.patch.object(p2p_checkpoint_manager, 'CheckpointManager', autospec=True) @mock.patch.object(p2p_options, 'CheckpointManagerOptions', autospec=True) + @mock.patch.object(p2p_checkpoint_manager, 'CheckpointManager', autospec=True) def test_create_checkpoint_manager( self, - mock_checkpoint_manager_options_cls, mock_checkpoint_manager_cls, + mock_checkpoint_manager_options_cls, ): local_dir = epath.Path('/tmp/local') persistent_dir = epath.Path('/tmp/persistent')