From 56b4630f737e22ea58aaac457679c52a7a9b4b3f Mon Sep 17 00:00:00 2001 From: Dref360 Date: Sat, 3 Sep 2022 16:44:04 -0400 Subject: [PATCH] Fix issue where pandas<1.4.0 doesn't return the number of rows --- src/datasets/io/sql.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/datasets/io/sql.py b/src/datasets/io/sql.py index d11a4ca70c0d..654d3589a1f8 100644 --- a/src/datasets/io/sql.py +++ b/src/datasets/io/sql.py @@ -114,7 +114,7 @@ def _write(self, conn: Connection, **to_sql_kwargs) -> int: df = self._batch_sql(offset) written += df.to_sql( self.table_name, conn, **to_sql_kwargs, if_exists="replace" if offset == 0 else "append" - ) + ) or len(df) else: num_rows, batch_size = len(self.dataset), self.batch_size