Skip to content

Add numpy argpartition function support #8732

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

Merged
sklam merged 5 commits into from
Feb 25, 2023
Merged

Add numpy argpartition function support #8732

Changes from 1 commit
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
5192cdb
Add numpy argpartition function support
Jan 28, 2023
193ec50
Change default parameter to None
Feb 7, 2023
413f2d5
Change logic based on default parameter
Feb 7, 2023
d5a66ed
Add boolean argpartition argument to partition factory
Feb 10, 2023
eefac1a
Fix line length error from flake8
Feb 11, 2023
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
Prev Previous commit
Fix line length error from flake8
  • Loading branch information
Rodrigo Agundez
Rodrigo Agundez committed Feb 11, 2023
commit eefac1a00702fd640e4f5674a91333b7d09f94f2
5 changes: 4 additions & 1 deletion numba/np/arraymath.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1524,7 +1524,10 @@ def _partition(A, low, high, I=None):

_partition = register_jitable(_partition_factory(less_than))
_partition_w_nan = register_jitable(_partition_factory(nan_aware_less_than))
_argpartition_w_nan = register_jitable(_partition_factory(nan_aware_less_than, argpartition=True))
_argpartition_w_nan = register_jitable(_partition_factory(
nan_aware_less_than,
argpartition=True)
)

Comment thread
rragundez marked this conversation as resolved.
Resolved

def _select_factory(partitionimpl):
Expand Down