Fix bucketing types (#719)

maxispeicher · jaidisido · web-flow · commit f1556917886a · 2021-05-27T20:17:40.000+01:00
Co-authored-by: jaidisido &lt;jaidisido@gmail.com&gt;
diff --git a/awswrangler/s3/_write_dataset.py b/awswrangler/s3/_write_dataset.py
@@ -81,7 +81,7 @@ def _to_buckets(
     **func_kwargs: Any,
 ) -> List[str]:
     _proxy: _WriteProxy = proxy if proxy else _WriteProxy(use_threads=False)
-    bucket_number_series = df.apply(
+    bucket_number_series = df.astype("O").apply(
         lambda row: _get_bucket_number(bucketing_info[1], [row[col_name] for col_name in bucketing_info[0]]),
         axis="columns",
     )

Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,7 @@ def _to_buckets(`
`81`	`81`	`**func_kwargs: Any,`
`82`	`82`	`) -> List[str]:`
`83`	`83`	`_proxy: _WriteProxy = proxy if proxy else _WriteProxy(use_threads=False)`
`84`		`- bucket_number_series = df.apply(`
	`84`	`+ bucket_number_series = df.astype("O").apply(`
`85`	`85`	`lambda row: _get_bucket_number(bucketing_info[1], [row[col_name] for col_name in bucketing_info[0]]),`
`86`	`86`	`axis="columns",`
`87`	`87`	`)`