Iter_batches does not work with a Snowflake connection in pl.read_database() #17404
Closed
2 tasks done
Labels
A-io-database
Area: reading/writing to databases
bug
Something isn't working
needs triage
Awaiting prioritization by a maintainer
python
Related to Python Polars
Checks
Reproducible example
Log output
Issue description
Iter_bacthes does not work with a snowflake connection in pl.read_database. It will throw a
TypeError TypeError: SnowflakeCursor.fetch_arrow_batches() takes 1 positional argument but 2 were given
. Looking at the snowflake connectior documentation it's probably because it does not support setting a batch_size parameter. However, this is required in pl.read_database if iter_batches is set to True. The connection works fine without iter_batches, so it seems it's a matter of not passing the batch_size if the connection is a snowflake type.Expected behavior
The expected behaviour would be for polars to return the batches from the snowflake connector.
Installed versions
The text was updated successfully, but these errors were encountered: