diff options
| author | Aaron Meurer <asmeurer@gmail.com> | 2021-01-12 12:57:45 -0700 |
|---|---|---|
| committer | Aaron Meurer <asmeurer@gmail.com> | 2021-01-12 12:57:45 -0700 |
| commit | e521b16844efc2853c0db9014098cb3e37f6eb04 (patch) | |
| tree | 6c55ff3828b608f00245c049b7cf89f5e4316eb6 /numpy/_array_api | |
| parent | 9578636259f86267c2253f4af2510ce1eeaf084c (diff) | |
| download | numpy-e521b16844efc2853c0db9014098cb3e37f6eb04.tar.gz | |
Add missing returns to the array API sorting functions
Diffstat (limited to 'numpy/_array_api')
| -rw-r--r-- | numpy/_array_api/_sorting_functions.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/numpy/_array_api/_sorting_functions.py b/numpy/_array_api/_sorting_functions.py index 98db3c7a2..fb2f819a2 100644 --- a/numpy/_array_api/_sorting_functions.py +++ b/numpy/_array_api/_sorting_functions.py @@ -6,6 +6,7 @@ def argsort(x, /, *, axis=-1, descending=False, stable=True): res = argsort(x, axis=axis, kind=kind) if descending: res = flip(res, axis=axis) + return res def sort(x, /, *, axis=-1, descending=False, stable=True): from .. import sort @@ -15,3 +16,4 @@ def sort(x, /, *, axis=-1, descending=False, stable=True): res = sort(x, axis=axis, kind=kind) if descending: res = flip(res, axis=axis) + return res |