diff options
Diffstat (limited to 'numpy/_array_api/_elementwise_functions.py')
| -rw-r--r-- | numpy/_array_api/_elementwise_functions.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/numpy/_array_api/_elementwise_functions.py b/numpy/_array_api/_elementwise_functions.py index aa48f440c..3ca71b53e 100644 --- a/numpy/_array_api/_elementwise_functions.py +++ b/numpy/_array_api/_elementwise_functions.py @@ -190,6 +190,9 @@ def ceil(x: array, /) -> array: """ if x.dtype not in _numeric_dtypes: raise TypeError('Only numeric dtypes are allowed in ceil') + if x.dtype in _integer_dtypes: + # Note: The return dtype of ceil is the same as the input + return x return ndarray._new(np.ceil(x._array)) def cos(x: array, /) -> array: @@ -258,6 +261,9 @@ def floor(x: array, /) -> array: """ if x.dtype not in _numeric_dtypes: raise TypeError('Only numeric dtypes are allowed in floor') + if x.dtype in _integer_dtypes: + # Note: The return dtype of floor is the same as the input + return x return ndarray._new(np.floor(x._array)) def floor_divide(x1: array, x2: array, /) -> array: |