Skip to content

Commit 04ac828

Browse files
committed
fix: test_torch_diff
1 parent dd1ab80 commit 04ac828

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

ivy_tests/test_ivy/test_frontends/test_torch/test_miscellaneous_ops.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1033,17 +1033,26 @@ def test_torch_diagonal(
10331033
min_num_dims=1,
10341034
valid_axis=True,
10351035
force_int_axis=True,
1036+
min_value=-1e03,
1037+
max_value=1e03,
1038+
abs_smallest_val=1e-03,
10361039
),
10371040
n=st.integers(min_value=0, max_value=5),
10381041
dtype_prepend=helpers.dtype_and_values(
10391042
available_dtypes=st.shared(helpers.get_dtypes("valid"), key="dtype"),
10401043
min_num_dims=1,
10411044
max_num_dims=1,
1045+
min_value=-1e03,
1046+
max_value=1e03,
1047+
abs_smallest_val=1e-03,
10421048
),
10431049
dtype_append=helpers.dtype_and_values(
10441050
available_dtypes=st.shared(helpers.get_dtypes("valid"), key="dtype"),
10451051
min_num_dims=1,
10461052
max_num_dims=1,
1053+
min_value=-1e03,
1054+
max_value=1e03,
1055+
abs_smallest_val=1e-03,
10471056
),
10481057
)
10491058
def test_torch_diff(
@@ -1058,10 +1067,10 @@ def test_torch_diff(
10581067
fn_tree,
10591068
):
10601069
input_dtype, x, axis = dtype_n_x_n_axis
1061-
_, prepend = dtype_prepend
1062-
_, append = dtype_append
1070+
prepend_dtype, prepend = dtype_prepend
1071+
append_dtype, append = dtype_append
10631072
helpers.test_frontend_function(
1064-
input_dtypes=input_dtype,
1073+
input_dtypes=[input_dtype[0], prepend_dtype[0], append_dtype[0]],
10651074
backend_to_test=backend_fw,
10661075
test_flags=test_flags,
10671076
frontend=frontend,
@@ -1071,6 +1080,9 @@ def test_torch_diff(
10711080
dim=axis,
10721081
prepend=prepend[0],
10731082
append=append[0],
1083+
atol=1e-03,
1084+
rtol=1e-03,
1085+
test_dtypes=False,
10741086
)
10751087

10761088

0 commit comments

Comments
 (0)