fix: Fixed ivy.argsort for jax backend (ivy-llc#28422)

Kacper-W-Kozdon · Feb 27, 2024 · 4196322 · 4196322
1 parent 9563832
commit 4196322
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy/functional/backends/jax/sorting.py b/ivy/functional/backends/jax/sorting.py
@@ -20,7 +20,7 @@ def argsort(
 ) -> JaxArray:
     kind = "stable" if stable else "quicksort"
     return (
-        jnp.argsort(-x, axis=axis, kind=kind)
+        jnp.argsort(x, axis=axis, kind=kind, descending=descending)
         if descending
         else jnp.argsort(x, axis=axis, kind=kind)
     )