Use nan_to_num instead of _replace_nan in nansum and nanprod

IntelPython · Feb 28, 2025 · 4c0908b · 4c0908b
1 parent 2bb1e8b
commit 4c0908b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dpnp/dpnp_iface_nanfunctions.py b/dpnp/dpnp_iface_nanfunctions.py
@@ -884,7 +884,7 @@ def nanprod(
 
     """
 
-    a, _ = _replace_nan(a, 1)
+    a = dpnp.nan_to_num(a, nan=1.0, posinf=dpnp.inf, neginf=-dpnp.inf)
     return dpnp.prod(
         a,
         axis=axis,
@@ -988,7 +988,7 @@ def nansum(
 
     """
 
-    a, _ = _replace_nan(a, 0)
+    a = dpnp.nan_to_num(a, nan=0.0, posinf=dpnp.inf, neginf=-dpnp.inf)
     return dpnp.sum(
         a,
         axis=axis,