From 026c41475e0550a909e36e5aa96ddf3713701869 Mon Sep 17 00:00:00 2001 From: Jackson McClintock Date: Wed, 7 Feb 2024 17:43:20 -0500 Subject: [PATCH] fix: Fix Frontend Failing Test: torch - tensor.torch.Tensor.__gt__ (#28219) --- .../test_frontends/test_torch/test_tensor.py | 38 ++++++++----------- 1 file changed, 16 insertions(+), 22 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 index b5490c5ac5f5e..9b04b1118ba20 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 @@ -752,28 +752,22 @@ def test_torch___gt__( backend_fw, ): input_dtype, x = dtype_and_x - try: - helpers.test_frontend_method( - init_input_dtypes=input_dtype, - backend_to_test=backend_fw, - init_all_as_kwargs_np={ - "data": x[0], - }, - method_input_dtypes=input_dtype, - method_all_as_kwargs_np={ - "other": x[1], - }, - frontend_method_data=frontend_method_data, - init_flags=init_flags, - method_flags=method_flags, - frontend=frontend, - on_device=on_device, - ) - except RuntimeError as e: - if "overflow" in e: - assume(False) - else: - raise + helpers.test_frontend_method( + init_input_dtypes=[input_dtype[0]], + backend_to_test=backend_fw, + init_all_as_kwargs_np={ + "data": x[0], + }, + method_input_dtypes=[input_dtype[1]], + method_all_as_kwargs_np={ + "other": x[1], + }, + frontend_method_data=frontend_method_data, + init_flags=init_flags, + method_flags=method_flags, + frontend=frontend, + on_device=on_device, + ) # __invert__