Skip to content

Commit

Permalink
test: exclude 'nearest_jax' interpolation case in frontend test of to…
Browse files Browse the repository at this point in the history
…rch quantile as it is invalid for torch.quantile, and run the test with 'nearest' interpolation.
  • Loading branch information
Ishticode committed Jan 6, 2024
1 parent 207dec0 commit bd26a28
Showing 1 changed file with 2 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -846,6 +846,8 @@ def test_torch_quantile(
input_dtype, x, axis, interpolation, q = dtype_and_x
if type(axis) is tuple:
axis = axis[0]
if interpolation == "nearest_jax":
interpolation = "nearest"
helpers.test_frontend_function(
input_dtypes=input_dtype,
backend_to_test=backend_fw,
Expand Down

0 comments on commit bd26a28

Please sign in to comment.