fix: bucketing error with newer version of Modin (0.31.0) (#2896)

aws · Jul 15, 2024 · ec1d7d6 · ec1d7d6
1 parent 2a4234a
commit ec1d7d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/awswrangler/s3/_write_dataset.py b/awswrangler/s3/_write_dataset.py
@@ -28,7 +28,7 @@ def _get_bucketing_series(df: pd.DataFrame, bucketing_info: typing.BucketingInfo
             axis="columns",
         )
     )
-    return bucket_number_series.astype(pd.CategoricalDtype(range(bucketing_info[1])))
+    return bucket_number_series.astype(np.array([pd.CategoricalDtype(range(bucketing_info[1]))]))
 
 
 def _simulate_overflow(value: int, bits: int = 31, signed: bool = False) -> int: