mariosasko / datasets_sql

An extension package of 🤗 Datasets that provides support for executing arbitrary SQL queries on HF datasets
Apache License 2.0
31 stars 2 forks source link

The readme demos are broken #1

Closed mo6zes closed 2 years ago

mo6zes commented 2 years ago

I tried running an example from the repo but the code is broken:

imdb_dset = load_dataset("imdb", split="train")
dataset = query(
    "SELECT text FROM imdb_dset"
)

results in AttributeError: 'duckdb.DuckDBPyConnection' object has no attribute 'fetch_arrow_chunk'

I am using datasets_sql version 0.1.1 and datasets version 2.5.2

mariosasko commented 2 years ago

Hi, thanks for reporting! I've just publised a new release of the package that fixes this issue.