From fb3307a8730249997578495dd334bbc0cebc7e8c Mon Sep 17 00:00:00 2001 From: Hana Joo Date: Tue, 7 Jul 2026 06:12:04 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 943851221 --- python/array_record_data_source.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/array_record_data_source.py b/python/array_record_data_source.py index 8166955..0eb7db6 100644 --- a/python/array_record_data_source.py +++ b/python/array_record_data_source.py @@ -140,7 +140,7 @@ def _run_in_parallel( # `executor.shutdown(cancel_futures=True)`. for remaining_future in thread_futures: remaining_future.cancel() - raise completed_future.exception() + raise completed_future.exception() # pyrefly: ignore[bad-raise] return [future.result() for future in thread_futures] @@ -442,7 +442,7 @@ def __init__( self._read_instructions = _get_read_instructions(paths) self._paths = [ri.filename for ri in self._read_instructions] self._reader_pool_size = ( - reader_pool_size or _get_flag_value(_ARRAY_RECORD_READER_POOL_SIZE) or 1 + reader_pool_size or _get_flag_value(_ARRAY_RECORD_READER_POOL_SIZE) or 1 # pyrefly: ignore[bad-argument-type] ) # Lock-free connection pool per shard