Skip to content

Commit ff68bca

Browse files
committed
validate that invert kwarg is boolean
1 parent 62a9fa4 commit ff68bca

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

dpctl/tensor/_set_functions.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -698,6 +698,9 @@ def isin(
698698
du.validate_usm_type(res_usm_type, allow_none=False)
699699
sycl_dev = exec_q.sycl_device
700700

701+
if invert not in [True, False]:
702+
raise ValueError(f"`invert` must be `True` or `False`, got {invert}")
703+
701704
x_dt = _get_dtype(x, sycl_dev)
702705
test_dt = _get_dtype(test_elements, sycl_dev)
703706
if not all(_validate_dtype(dt) for dt in (x_dt, test_dt)):

0 commit comments

Comments
 (0)