fix: ensure traced function returns are converted to ivy in testing (#…

…28494)
ivy-llc · Mar 7, 2024 · da79905 · da79905
1 parent 578e88d
commit da79905
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ivy_tests/test_ivy/helpers/function_testing.py b/ivy_tests/test_ivy/helpers/function_testing.py
@@ -2580,6 +2580,8 @@ def get_ret_and_flattened_np_array(
     with BackendHandler.update_backend(backend_to_test) as ivy_backend:
         with ivy_backend.PreciseMode(precision_mode):
             ret = fn(*args, **kwargs)
+            if test_trace or test_trace_each:
+                ret = ivy_backend.to_ivy(ret, True)
 
         def map_fn(x):
             if _is_frontend_array(x):