Skip to content

Commit 4989a7d

Browse files
committed
Specify that Python scalars are permitted for max and min in clip
1 parent 56e98b2 commit 4989a7d

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

dpctl/tensor/_clip.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -399,10 +399,12 @@ def clip(x, /, min=None, max=None, out=None, order="K"):
399399
x (usm_ndarray): Array containing elements to clip.
400400
Must be compatible with `min` and `max` according
401401
to broadcasting rules.
402-
min ({None, usm_ndarray}, optional): Array containing minimum values.
402+
min ({None, Union[usm_ndarray, bool, int, float, complex]}, optional):
403+
Array containing minimum values.
403404
Must be compatible with `x` and `max` according
404405
to broadcasting rules.
405-
max ({None, usm_ndarray}, optional): Array containing maximum values.
406+
max ({None, Union[usm_ndarray, bool, int, float, complex]}, optional):
407+
Array containing maximum values.
406408
Must be compatible with `x` and `min` according
407409
to broadcasting rules.
408410
out ({None, usm_ndarray}, optional):

0 commit comments

Comments
 (0)