fix paddle.nanmedian frontend function (ivy-llc#23578)

samthakur587 · Sep 18, 2023 · cc0e4ea · cc0e4ea
1 parent 200255b
commit cc0e4ea
Showing 1 changed file with 2 additions and 7 deletions.
diff --git 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")