File tree Expand file tree Collapse file tree 3 files changed +9
-2
lines changed Expand file tree Collapse file tree 3 files changed +9
-2
lines changed Original file line number Diff line number Diff line change @@ -58,6 +58,7 @@ def arrow_table_to_pandas(
58
58
) -> pd .DataFrame :
59
59
pa = import_optional_dependency ("pyarrow" )
60
60
61
+ types_mapper : type [pd .ArrowDtype ] | None | Callable
61
62
if dtype_backend == "numpy_nullable" :
62
63
mapping = _arrow_dtype_mapping ()
63
64
if null_to_int64 :
Original file line number Diff line number Diff line change @@ -2210,6 +2210,9 @@ def read_table(
2210
2210
with self .con .cursor () as cur :
2211
2211
cur .execute (stmt )
2212
2212
pa_table = cur .fetch_arrow_table ()
2213
+ dtype_backend = (
2214
+ lib .no_default if dtype_backend == "numpy" else dtype_backend
2215
+ )
2213
2216
df = arrow_table_to_pandas (pa_table , dtype_backend = dtype_backend )
2214
2217
2215
2218
return _wrap_result_adbc (
@@ -2281,6 +2284,9 @@ def read_query(
2281
2284
with self .con .cursor () as cur :
2282
2285
cur .execute (sql )
2283
2286
pa_table = cur .fetch_arrow_table ()
2287
+ dtype_backend = (
2288
+ lib .no_default if dtype_backend == "numpy" else dtype_backend
2289
+ )
2284
2290
df = arrow_table_to_pandas (pa_table , dtype_backend = dtype_backend )
2285
2291
2286
2292
return _wrap_result_adbc (
Original file line number Diff line number Diff line change @@ -959,12 +959,12 @@ def sqlite_buildin_types(sqlite_buildin, types_data):
959
959
960
960
adbc_connectable_iris = [
961
961
pytest .param ("postgresql_adbc_iris" , marks = pytest .mark .db ),
962
- pytest . param ( "sqlite_adbc_iris" , marks = pytest . mark . db ) ,
962
+ "sqlite_adbc_iris" ,
963
963
]
964
964
965
965
adbc_connectable_types = [
966
966
pytest .param ("postgresql_adbc_types" , marks = pytest .mark .db ),
967
- pytest . param ( "sqlite_adbc_types" , marks = pytest . mark . db ) ,
967
+ "sqlite_adbc_types" ,
968
968
]
969
969
970
970
You can’t perform that action at this time.
0 commit comments