handle when axes is list for rot90 in jax backend (ivy-llc#10513)

vedpatwardhan · Feb 13, 2023 · 29a0d3d · 29a0d3d
1 parent 26b57e0
commit 29a0d3d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ivy/functional/backends/jax/experimental/manipulation.py b/ivy/functional/backends/jax/experimental/manipulation.py
@@ -74,6 +74,8 @@ def rot90(
     axes: Optional[Tuple[int, int]] = (0, 1),
     out: Optional[JaxArray] = None,
 ) -> JaxArray:
+    if isinstance(axes, list):
+        axes = tuple(axes)
     return jnp.rot90(m, k, axes)