From bea5442171cdbacccb3b8aa5185e41b5a86b5ecf Mon Sep 17 00:00:00 2001 From: jack-xia-dp <144161208+jackx111@users.noreply.github.com> Date: Mon, 7 Oct 2024 11:16:58 -0400 Subject: [PATCH] save --- howso/utilities/utilities.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/howso/utilities/utilities.py b/howso/utilities/utilities.py index 881c9d27..68e7efaf 100644 --- a/howso/utilities/utilities.py +++ b/howso/utilities/utilities.py @@ -1397,7 +1397,7 @@ def format_confusion_matrix(confusion_matrix: dict[str, dict[str, int]]) -> tupl def yield_dataframe_as_chunks(df: pd.DataFrame, num_chunks: int) -> t.Generator[pd.DataFrame, None, None]: """ - Splits and yields a DataFrame in chunks using iloc. Np.array_split is deprecated. + Yields a DataFrame in chunks using iloc. Np.array_split is deprecated. Parameters ---------- @@ -1420,7 +1420,7 @@ def yield_dataframe_as_chunks(df: pd.DataFrame, num_chunks: int) -> t.Generator[ ) yield df - rows_per_chunk = ceil(total_rows // num_chunks) + rows_per_chunk = ceil(total_rows / num_chunks) for i in range(num_chunks): start = i * rows_per_chunk