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): Add to_jax methods to support Jax Array export from DataFrame and Series #16294

Merged
merged 11 commits into from
May 19, 2024
Merged

feat(python): Add to_jax methods to support Jax Array export from DataFrame and Series #16294

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
b653125
feat(python): Add new `to_jax` method to support export to jax arrays…
alexander-beedie May 16, 2024
4789e13
minor docstring update
alexander-beedie May 17, 2024
edd2bec
skip a test under py3.8
alexander-beedie May 17, 2024
a75560e
include a `versionadded` tag
alexander-beedie May 17, 2024
5319780
bonus test coverage
alexander-beedie May 18, 2024
7bf53dd
fix default `JAX_ENABLE_X64` env value
alexander-beedie May 18, 2024
d9f1386
make use of `pytestmark`
alexander-beedie May 19, 2024
6eef9b9
mark as `unstable`
alexander-beedie May 19, 2024
1e41760
add `to_jax` entry to the sphinx docs
alexander-beedie May 19, 2024
a68782a
add `series` sphinx docs entry and mark `unstable`
alexander-beedie May 19, 2024
362b8d9
add a docstring note to series `to_torch`
alexander-beedie May 19, 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
2 changes: 1 addition & 1 deletion crates/polars-core/src/chunked_array/ops/aggregate/mod.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -195,7 +195,7 @@ where
}
}

/// Booleans are casted to 1 or 0.
/// Booleans are cast to 1 or 0.
impl BooleanChunked {
pub fn sum(&self) -> Option<IdxSize> {
Some(if self.is_empty() {
Expand Down
1 change: 1 addition & 0 deletions py-polars/docs/source/reference/dataframe/export.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ Export DataFrame data to other formats:
DataFrame.to_dict
DataFrame.to_dicts
DataFrame.to_init_repr
DataFrame.to_jax
DataFrame.to_numpy
DataFrame.to_pandas
DataFrame.to_struct
Expand Down
2 changes: 2 additions & 0 deletions py-polars/docs/source/reference/series/export.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,9 @@ Export Series data to other formats:

Series.to_arrow
Series.to_frame
Series.to_jax
Series.to_list
Series.to_numpy
Series.to_pandas
Series.to_init_repr
Series.to_torch
Loading