From 9788f572a75c7193371e9bf06b4c6a8cd3aa99af Mon Sep 17 00:00:00 2001 From: Justin Pan Date: Tue, 16 Jun 2026 18:07:04 -0700 Subject: [PATCH] Internal change. PiperOrigin-RevId: 933404006 --- .../orbax/checkpoint/_src/serialization/jax_array_handlers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/checkpoint/orbax/checkpoint/_src/serialization/jax_array_handlers.py b/checkpoint/orbax/checkpoint/_src/serialization/jax_array_handlers.py index 1f0a7ed09..d06e792d4 100644 --- a/checkpoint/orbax/checkpoint/_src/serialization/jax_array_handlers.py +++ b/checkpoint/orbax/checkpoint/_src/serialization/jax_array_handlers.py @@ -635,6 +635,7 @@ def _serialize_arrays( deprioritized = prioritized_async + deprioritized + if dispatcher is None: return _serialize_arrays_batches_without_dispatcher( prioritized,