fix: Fix broken module converter tests (#27089)

ivy-llc · Oct 23, 2023 · 0665a76 · 0665a76
1 parent 540534c
commit 0665a76
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)