Skip to content

Commit 0892940

Browse files
edwinsolisfsyurkevi
authored andcommitted
Fixed missing decorator on af.select
1 parent 875a398 commit 0892940

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

arrayfire/library/array_functions.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -954,6 +954,7 @@ def replace(lhs: Array, rhs: Array | int | float, conditional: Array, /) -> None
954954
wrapper.replace_scalar(lhs.arr, conditional.arr, rhs)
955955

956956

957+
@afarray_as_array
957958
def select(lhs: Array | int | float, rhs: Array | int | float, conditional: Array, /) -> Array:
958959
"""
959960
Conditionally selects elements from one of two sources (ArrayFire arrays or scalars) based on a condition array.

0 commit comments

Comments
 (0)