Merge branch 'main' into dependabot/pip/pytest-8.0.0

Author: ya7010

turu-snowflake/src/turu/snowflake/record/async_record_cursor.py

@@ -39,7 +39,7 @@ async def fetch_pandas_batches(
         if isinstance(self._recorder, turu.core.record.CsvRecorder):
             if limit := self._recorder._options.get("limit"):
                 async for batch in batches:
-                    yield batch
+                    yield batch.head(limit)
 
                     limit -= len(batch)
                     if limit <= 0:

turu-snowflake/src/turu/snowflake/record/record_cursor.py

@@ -35,7 +35,7 @@ def fetch_pandas_batches(self, **kwargs) -> Iterator[GenericPandasDataFrame]:
         if isinstance(self._recorder, turu.core.record.CsvRecorder):
             if limit := self._recorder._options.get("limit"):
                 for batch in batches:
-                    yield batch
+                    yield batch.head(limit)
 
                     limit -= len(batch)
                     if limit <= 0: