Merge pull request ivy-llc#10447 from clsandoval/torch.tensor.reshape…

….test.fix add fix for torch tensor reshape tests
vedpatwardhan · Feb 14, 2023 · 878a3a8 · 878a3a8
2 parents be31bd7 + 1d842bd
commit 878a3a8
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py b/ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py
@@ -387,17 +387,15 @@ def test_torch_instance_reshape(
     frontend,
 ):
     input_dtype, x = dtype_x
+    shape = {
+        "shape": shape,
+    }
     if unpack_shape:
-        method_flags.num_positional_args = len(shape) + 1
-        shape = {}
+        method_flags.num_positional_args = len(shape["shape"]) + 1
         i = 0
-        for x_ in shape:
+        for x_ in shape["shape"]:
             shape["x{}".format(i)] = x_
             i += 1
-    else:
-        shape = {
-            "shape": shape,
-        }
     helpers.test_frontend_method(
         init_input_dtypes=input_dtype,
         init_all_as_kwargs_np={