Fix module converter tests

ivy-llc · Oct 21, 2023 · 007a7fc · 007a7fc
1 parent a76e213
commit 007a7fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy/functional/ivy/gradients.py b/ivy/functional/ivy/gradients.py
@@ -109,7 +109,7 @@ def _get_required_float_variables(xs, xs_grad_idxs):
     Also, returns a list of duplicate index chains for the nested
     structure.
     """
-    if (ivy.is_ivy_container(xs) or ivy.is_array(xs)) and xs_grad_idxs == [[0]]:
+    if (ivy.is_ivy_container(xs) or ivy.is_array(xs)) and xs_grad_idxs == ((0,),):
         xs_grad_idxs = None
     duplicate_index_chains = _get_duplicate_index_chains(xs)
     xs = _to_ivy(xs)