Skip to content

Commit

Permalink
FIX! Fix Ivy Failing Test: torch - searching.argmax (#28117)
Browse files Browse the repository at this point in the history
Co-authored-by: NripeshN <nripesh14@gmail.com>
  • Loading branch information
MuhammadNizamani and NripeshN committed Feb 1, 2024
1 parent fff4830 commit 171d7aa
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ def _dtype_x_limited_axis(draw, *, allow_none=False):
fn_tree="functional.ivy.argmax",
dtype_x_axis=_dtype_x_limited_axis(allow_none=True),
keepdims=st.booleans(),
dtype=helpers.get_dtypes("valid", full=False, none=True),
dtype=helpers.get_dtypes("numeric", full=False, none=True),
select_last_index=st.booleans(),
)
def test_argmax(
Expand Down

0 comments on commit 171d7aa

Please sign in to comment.