Skip to content

Commit eec459e

Browse files
Chaluvadisyurkevi
authored andcommitted
fixed sum and product function logic
1 parent 79216c4 commit eec459e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

arrayfire/library/vector_algorithms.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -207,9 +207,9 @@ def sum(array: Array, /, *, axis: int | None = None, nan_value: float | None = N
207207
return wrapper.sum_nan_all(array.arr, nan_value)
208208

209209
if nan_value is not None:
210-
return Array.from_afarray(wrapper.sum(array.arr, axis))
210+
return Array.from_afarray(wrapper.sum_nan(array.arr, axis, nan_value))
211211

212-
return Array.from_afarray(wrapper.sum_nan(array.arr, axis, nan_value=nan_value)) # type: ignore[call-arg]
212+
return Array.from_afarray(wrapper.sum(array.arr, axis)) # type: ignore[call-arg]
213213

214214

215215
def product(
@@ -247,7 +247,7 @@ def product(
247247
if nan_value is None:
248248
return Array.from_afarray(wrapper.product(array.arr, axis))
249249

250-
return Array.from_afarray(wrapper.product_nan(array.arr, axis, nan_value=nan_value)) # type: ignore[call-arg]
250+
return Array.from_afarray(wrapper.product_nan(array.arr, axis, nan_value)) # type: ignore[call-arg]
251251

252252

253253
def count(

0 commit comments

Comments
 (0)