fix: Fix to handle_transpose_in_input_and_output_for_functions decora…

…tor to correctly propagate the currently set data format to subsequent calls and reset it once we exit from those calls

ivy/utils/decorator_utils.py

@@ -432,8 +432,11 @@ def transpose_wrapper(*args, **kwargs):
                 kwargs["data_format"] = value_map[kwargs["data_format"]]
             if "filter_format" in kwargs and kwargs["filter_format"] in value_map:
                 kwargs["filter_format"] = value_map[kwargs["filter_format"]]
+                os.environ["DATA_FORMAT"] = "channels_last"
 
-        return fn(*args, **kwargs)
+        res = fn(*args, **kwargs)
+        os.environ["DATA_FORMAT"] = DATA_FORMAT
+        return res
 
     return transpose_wrapper