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

ENH: stats.rankdata: add array API support #20639

Draft
wants to merge 4 commits into
base: main
Choose a base branch
from

Conversation

mdhaber
Copy link
Contributor

@mdhaber mdhaber commented May 3, 2024

Reference issue

Towards gh-20544

What does this implement/fix?

Adds array API support to scipy.stats.rankdata.

Additional information

No need to review this yet. While working on it, I found that it needs several things that aren't covered by array_api_compat yet (e.g. repeat, moveaxis, NumPy asarray with copy=False), and it would also really benefit from the array API supporting take_along_axis (data-apis/array-api#177) and diff (data-apis/array-api#791). Also, without put (or preferably put_along_axis), this needed a second argsort. As-is, this would introduce a substantial performance regression.

@mdhaber mdhaber added scipy.stats enhancement A new feature or improvement array types Items related to array API support and input array validation (see gh-18286) labels May 3, 2024
@github-actions github-actions bot added scipy._lib CI Items related to the CI tools such as CircleCI, GitHub Actions or Azure labels May 3, 2024
@mdhaber mdhaber mentioned this pull request May 8, 2024
68 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
array types Items related to array API support and input array validation (see gh-18286) CI Items related to the CI tools such as CircleCI, GitHub Actions or Azure enhancement A new feature or improvement scipy._lib scipy.stats
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant