From c30ee4b76643c3ff9ac8f50c77721a8a19df5dab Mon Sep 17 00:00:00 2001 From: Adam Reeve Date: Tue, 22 Sep 2026 22:19:02 +1200 Subject: [PATCH] Fix errors during test collection when Parquet encryption unavailable --- python/pyarrow/tests/parquet/test_encryption.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/pyarrow/tests/parquet/test_encryption.py b/python/pyarrow/tests/parquet/test_encryption.py index 6d0cfc0024e..5185bf6a301 100644 --- a/python/pyarrow/tests/parquet/test_encryption.py +++ b/python/pyarrow/tests/parquet/test_encryption.py @@ -680,9 +680,12 @@ def check_rotated_external_keys(master_key_id: str) -> None: assert data_table.equals(table_read_after_rotation) -def recording_kms_factory(created_configs, client_class=InMemoryKmsClient): +def recording_kms_factory(created_configs, client_class=None): """Create a KMS client factory that appends the KMS instance ID and URL of each connection configuration it is given to created_configs""" + if client_class is None: + client_class = InMemoryKmsClient + def kms_factory(kms_connection_configuration): created_configs.append( (kms_connection_configuration.kms_instance_id,