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.

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

feat: enable eq and neq for array dtype #12020

Merged
merged 1 commit into from
Oct 26, 2023
Merged

Conversation

reswqa
Copy link
Collaborator

@reswqa reswqa commented Oct 25, 2023

This closes #12012.

Due to another bug(#12012 (comment)), tests for null array can't introduce atm.

@github-actions github-actions bot added enhancement New feature or an improvement of an existing feature python Related to Python Polars rust Related to Rust Polars labels Oct 25, 2023
let mask = combine_validities_and(Some(mask.values()), mask.validity()).unwrap();
let (slice, offset, _len) = mask.as_slice();
assert_eq!(offset, 0);

let width = a.size();
let iter = (0..a.len()).map(|i| func(count_zeros(slice, i, width)));
let iter = (0..a.len()).map(|i| func(count_zeros(slice, i * width, width)));
Copy link
Collaborator Author

@reswqa reswqa Oct 25, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

IIUC, this should be a bug before. 🤔

@ritchie46
Copy link
Member

Nice one! @reswqa 🙌

@ritchie46 ritchie46 merged commit a81a404 into pola-rs:main Oct 26, 2023
26 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or an improvement of an existing feature python Related to Python Polars rust Related to Rust Polars
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Implement equality for Array type columns
2 participants