From cc0e4ea4cf4a77f006e4389ce8f7fb35c2869266 Mon Sep 17 00:00:00 2001 From: Ahmed Hisham <62694574+Mghrabi@users.noreply.github.com> Date: Mon, 18 Sep 2023 21:24:35 +0300 Subject: [PATCH] fix paddle.nanmedian frontend function (#23578) --- ivy/functional/frontends/paddle/stat.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/ivy/functional/frontends/paddle/stat.py b/ivy/functional/frontends/paddle/stat.py index 91b69d7441091..83b4d5131e556 100644 --- a/ivy/functional/frontends/paddle/stat.py +++ b/ivy/functional/frontends/paddle/stat.py @@ -28,17 +28,12 @@ def median(x, axis=None, keepdim=False, name=None): @with_supported_dtypes( - {"2.5.0 and below": ("float16", "float32", "float64", "uint16")}, + {"2.5.0 and below": ("float16", "float32", "float64", "int32", "int64")}, "paddle", ) @to_ivy_arrays_and_back def nanmedian(x, axis=None, keepdim=True, name=None): - x = ( - ivy.astype(x, ivy.float64) - if ivy.dtype(x) == "float64" - else ivy.astype(x, ivy.float32) - ) - return ivy.median(x, axis=axis, keepdims=keepdim) + return ivy.nanmedian(x, axis=axis, keepdims=keepdim) @with_unsupported_dtypes({"2.5.1 and below": ("complex", "int8")}, "paddle")