From eed8ebdb05943c4d6618b4ee422e09ec1c271635 Mon Sep 17 00:00:00 2001 From: aibenStunner Date: Sun, 18 Feb 2024 20:09:02 +0000 Subject: [PATCH] fix: failing test for torch tensor repeat --- ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py | 6 ++++++ 1 file changed, 6 insertions(+) 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 index 26e8c5ece6238..83802016e256b 100644 --- a/ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py +++ b/ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py @@ -11344,6 +11344,12 @@ def test_torch_repeat( backend_fw, ): input_dtype, x, repeats = dtype_x_repeats + + if backend_fw == "paddle": + # paddle only supports size of the shape of repeats + # to be less than or equal to 6 + assume(len(repeats) <= 6) + repeat = { "repeats": repeats, }