ENH: add `partition` and `argpartition` functions - #449

Merged

lucascolley merged 15 commits into

data-apis:mainfrom

cakedev0:partition

Oct 3, 2025

Commits on Oct 1, 2025

initial work with a small test; fails for sparse backend as it has not argsort;
cakedev0
committed
51ade21
Support for multi-dimensional arrays
cakedev0
committed
81b8ac3
Test input validation
cakedev0
committed
45121c5

Commits on Oct 2, 2025

adress PR comments
cakedev0
committed
74c509f
improved tests & coverage
cakedev0
committed
6efc73a
fix xfail thingy
cakedev0
committed
6e69083
add in err msg: no argsort
cakedev0
committed
259f93d
Docstring improvements
cakedev0
committed
c88e93e
rewrite of the torch logic
cakedev0
committed
579b3bc
dask support in argpart tests
cakedev0
committed
c2827da
Merge remote-tracking branch 'upstream/main' into partition
cakedev0
committed
8d5af47

Commits on Oct 3, 2025

Fix docstring format

cakedev0
and
lucascolley
authored
b2a567e
numpy doc: notes format
cakedev0
committed
ac19a23
Merge remote-tracking branch 'upstream/main' into partition
cakedev0
committed
2a3fd9f
comment about peak memory
cakedev0
committed
1084835