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

ivy-llc · Feb 26, 2024 · e22cd90 · e22cd90
1 parent e1c93cb
commit e22cd90
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)
     )