Skip to content

Commit 1e687d5

Browse files
sakchalsyurkevi
authored andcommitted
changed sum function logic
1 parent eec459e commit 1e687d5

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
@@ -206,10 +206,10 @@ def sum(array: Array, /, *, axis: int | None = None, nan_value: float | None = N
206206

207207
return wrapper.sum_nan_all(array.arr, nan_value)
208208

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

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

214214

215215
def product(

0 commit comments

Comments
 (0)