From 77f2dc258874f8e53654026dffb61d8d1b5a6ddb Mon Sep 17 00:00:00 2001 From: jaidisido Date: Tue, 5 Mar 2024 14:12:48 +0000 Subject: [PATCH] fix: indent categories correctly in pyarrow_additional_kwargs (#2701) Co-authored-by: Leon Luttenberger --- awswrangler/athena/_read.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/awswrangler/athena/_read.py b/awswrangler/athena/_read.py index 6ca2f7c47..21b502b4d 100644 --- a/awswrangler/athena/_read.py +++ b/awswrangler/athena/_read.py @@ -149,8 +149,8 @@ def _fetch_parquet_result( return df if not pyarrow_additional_kwargs: pyarrow_additional_kwargs = {} - if categories: - pyarrow_additional_kwargs["categories"] = categories + if categories: + pyarrow_additional_kwargs["categories"] = categories _logger.debug("Reading Parquet result from %d paths", len(paths)) ret = s3.read_parquet( path=paths,