Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

feat(python): Support PyCapsule Interface in DataFrame & Series constructors #17693

Merged
merged 12 commits into from
Jul 25, 2024
Merged

feat(python): Support PyCapsule Interface in DataFrame & Series constructors #17693

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
a0e10be
Support __arrow_c_stream__ in DataFrame constructor
kylebarron Jul 18, 2024
9754fb1
Support stream import to series and use that for DataFrame import
kylebarron Jul 18, 2024
d0ea973
Import __arrow_c_array__ as well
kylebarron Jul 18, 2024
f4f1162
Merge branch 'main' into kyle/c-stream-import
kylebarron Jul 21, 2024
763e153
reorder pycapsule interface checks
kylebarron Jul 21, 2024
6c4de77
remove pandas not installed check
kylebarron Jul 21, 2024
7446467
Add typing for C Data and Stream protocols
kylebarron Jul 21, 2024
b576372
add constructor tests
kylebarron Jul 21, 2024
477ed23
add test cases for empty arrays/streams
kylebarron Jul 21, 2024
1bfb479
Add safety docs
kylebarron Jul 23, 2024
1618d1d
Test for pandas import via pycapsule interface
kylebarron Jul 23, 2024
9fefdeb
merge
ritchie46 Jul 25, 2024
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
5 changes: 0 additions & 5 deletions py-polars/tests/unit/constructors/test_constructors.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -958,11 +958,6 @@ def _constructor(self) -> type:
assert df.schema == {"colx": pl.Datetime("us")}
assert df.rows() == [(datetime(2022, 10, 31, 10, 30, 45, 123456),)]

# pandas is not available
monkeypatch.setattr(pl.dataframe.frame, "_check_for_pandas", lambda x: False)
with pytest.raises(TypeError):
pl.DataFrame(pandas_df)

kylebarron marked this conversation as resolved.
Show resolved Hide resolved

def test_init_errors() -> None:
# Length mismatch
Expand Down