diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2022-01-12 13:12:54 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-01-12 13:12:54 -0700 |
| commit | f98c60a035b007680fe811eeed3b6326961ccff5 (patch) | |
| tree | 73556ad843b2ac6670bb2840d8eb4982c70d171e /numpy/array_api/_elementwise_functions.py | |
| parent | f637d75eb6cbcc90ceadee3e62e6b62fd0e53976 (diff) | |
| parent | e3406ed3ef83f7de0f3419361e85bd8634d0fd2b (diff) | |
| download | numpy-f98c60a035b007680fe811eeed3b6326961ccff5.tar.gz | |
Merge pull request #20762 from honno/xp-pow-fixes
BUG: Allow integer inputs for pow-related functions in `array_api`
Diffstat (limited to 'numpy/array_api/_elementwise_functions.py')
| -rw-r--r-- | numpy/array_api/_elementwise_functions.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/array_api/_elementwise_functions.py b/numpy/array_api/_elementwise_functions.py index 4408fe833..c758a0944 100644 --- a/numpy/array_api/_elementwise_functions.py +++ b/numpy/array_api/_elementwise_functions.py @@ -591,8 +591,8 @@ def pow(x1: Array, x2: Array, /) -> Array: See its docstring for more information. """ - if x1.dtype not in _floating_dtypes or x2.dtype not in _floating_dtypes: - raise TypeError("Only floating-point dtypes are allowed in pow") + if x1.dtype not in _numeric_dtypes or x2.dtype not in _numeric_dtypes: + raise TypeError("Only numeric dtypes are allowed in pow") # Call result type here just to raise on disallowed type combinations _result_type(x1.dtype, x2.dtype) x1, x2 = Array._normalize_two_args(x1, x2) |