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")