diff --git a/python/ray/air/util/tensor_extensions/arrow.py b/python/ray/air/util/tensor_extensions/arrow.py
index a101ddd61cd5..08861896b1f3 100644
--- a/python/ray/air/util/tensor_extensions/arrow.py
+++ b/python/ray/air/util/tensor_extensions/arrow.py
@@ -127,6 +127,8 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
 
     def __arrow_ext_serialize__(self) -> bytes:
         """Serialize the extension type using caching if enabled."""
+        if self._serialize_cache is not None:
+            return self._serialize_cache
         with self._cache_lock:
             if self._serialize_cache is None:
                 self._serialize_cache = self._arrow_ext_serialize_compute()