From d288ea4720789e97c213978e0987cea83816b42c Mon Sep 17 00:00:00 2001 From: Mohammed Ayman Date: Sat, 19 Aug 2023 15:36:57 +0300 Subject: [PATCH] minimal change to frontend torch pooling functions tests (#22185) Co-authored-by: Mohammed --- .../test_nn/test_functional/test_pooling_functions.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ivy_tests/test_ivy/test_frontends/test_torch/test_nn/test_functional/test_pooling_functions.py b/ivy_tests/test_ivy/test_frontends/test_torch/test_nn/test_functional/test_pooling_functions.py index d781e8c0dd804..cb7c27a1ce5a7 100644 --- a/ivy_tests/test_ivy/test_frontends/test_torch/test_nn/test_functional/test_pooling_functions.py +++ b/ivy_tests/test_ivy/test_frontends/test_torch/test_nn/test_functional/test_pooling_functions.py @@ -68,7 +68,7 @@ def test_torch_avg_pool1d( # TODO: remove the processing of padding attribute when ivy.avg_pool # support explicit padding x_shape = [x[0].shape[2]] - padding = (pad[i] for i, pad in enumerate(padding)) + padding = [pad[i] for i, pad in enumerate(padding)] # figuring out the exact kernel_size for SAME and VALID padding # As ivy.avg_pool1d doesn't support explicit padding scheme if not sum(padding) == 0: @@ -118,7 +118,7 @@ def test_torch_avg_pool2d( input_dtype, x, kernel_size, stride, padding = dtype_x_k_s # TODO: remove the processing of padding attribute when ivy.avg_pool # support explicit padding - padding = (pad[i] for i, pad in enumerate(padding)) + padding = [pad[i] for i, pad in enumerate(padding)] x_shape = x[0].shape[2:] if not sum(padding) == 0: padding = calculate_same_padding(kernel_size, [stride[0]] * 2, x_shape)