fix: Fixed passing of missing ret_idxs argument in `_rebuild_flatte…

…ned_containers()` function call (ivy-llc#27822)
Mac16661 · Jan 3, 2024 · ef2c6d0 · ef2c6d0
1 parent 086445a
commit ef2c6d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy/functional/backends/mxnet/experimental/gradients.py b/ivy/functional/backends/mxnet/experimental/gradients.py
@@ -46,7 +46,7 @@ def vjpfun(x_in):
         )
 
         return _rebuild_flattened_containers(
-            ivy.to_ivy(grads, nested=True, include_derived=True)
+            ivy.to_ivy(grads, nested=True, include_derived=True), ret_idxs
         )
 
     return (ivy.to_ivy(primals_out, nested=True, include_derived=True), vjpfun)