Tests passing locally, ready for CB submission

Author: jaidisido

awswrangler/s3/_read_parquet.py

@@ -246,7 +246,7 @@ def _read_parquet_chunked(
     use_threads: Union[bool, int],
     arrow_kwargs: Dict[str, Any],
 ) -> Iterator[pd.DataFrame]:
-    batch_size = None if isinstance(chunked, bool) else chunked
+    batch_size = None if chunked is True else chunked
     tables = _utils.batches_to_table(
         pieces=pieces,
         schema=schema,
@@ -255,8 +255,24 @@ def _read_parquet_chunked(
         use_threads=use_threads,
         batch_size=batch_size,
     )
+
+    next_slice: Optional[pd.DataFrame] = None
     for table in tables:
-        yield _table_to_df(table=table, kwargs=arrow_kwargs)
+        df = _table_to_df(table=table, kwargs=arrow_kwargs)
+        if chunked is True:
+            yield df
+        else:
+            if next_slice is not None:
+                df = pd.concat(objs=[next_slice, df], sort=False, copy=False)
+            while len(df.index) >= chunked:
+                yield df.iloc[:chunked, :].copy()
+                df = df.iloc[chunked:, :]
+            if df.empty:
+                next_slice = None
+            else:
+                next_slice = df
+    if next_slice is not None:
+        yield next_slice
 
 
 def _read_parquet(