Skip to content
This repository has been archived by the owner on Mar 18, 2022. It is now read-only.

return pyarrow.Table instead of return the pandas-compatible NumPy ar… #319

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

return pyarrow.Table instead of return the pandas-compatible NumPy ar… #319

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 3 additions & 5 deletions dremio_client/flight/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -146,11 +146,9 @@ def query(
batches.append(batch)
except StopIteration:
break
data = pa.Table.from_batches(batches)
if pandas:
return data.to_pandas()
else:
return data
table = pa.Table.from_batches(batches)

return table


except ImportError:
Expand Down