@@ -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)
10491058def 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