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

LeonLuttenberger · web-flow · commit ec1d7d68226b · 2024-07-15T10:02:52.000+01:00
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:

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ def _get_bucketing_series(df: pd.DataFrame, bucketing_info: typing.BucketingInfo`
`28`	`28`	`axis="columns",`
`29`	`29`	`)`
`30`	`30`	`)`
`31`		`- return bucket_number_series.astype(pd.CategoricalDtype(range(bucketing_info[1])))`
	`31`	`+ return bucket_number_series.astype(np.array([pd.CategoricalDtype(range(bucketing_info[1]))]))`
`32`	`32`
`33`	`33`
`34`	`34`	`def _simulate_overflow(value: int, bits: int = 31, signed: bool = False) -> int:`