fix: fixed frontend test for torch.Tensor.repeat (ivy-llc#28583)

by putting a limit on array max dims and max size
ogbanugot · Mar 16, 2024 · e0791d1 · e0791d1
1 parent dc171be
commit e0791d1
Showing 1 changed file with 9 additions and 1 deletion.
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
@@ -347,7 +347,15 @@ def _repeat_helper(draw):
         )
     )
 
-    repeats = draw(st.lists(st.integers(min_value=1, max_value=5), min_size=len(shape)))
+    MAX_NUMPY_DIMS = 32
+    repeats = draw(
+        st.lists(
+            st.integers(min_value=1, max_value=5),
+            min_size=len(shape),
+            max_size=MAX_NUMPY_DIMS,
+        )
+    )
+    assume(np.prod(repeats) * np.prod(shape) <= 2**28)
     return input_dtype, x, repeats