diff options
Diffstat (limited to 'numpy/_array_api/_set_functions.py')
| -rw-r--r-- | numpy/_array_api/_set_functions.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/numpy/_array_api/_set_functions.py b/numpy/_array_api/_set_functions.py index fd1438be5..80288c57d 100644 --- a/numpy/_array_api/_set_functions.py +++ b/numpy/_array_api/_set_functions.py @@ -1,6 +1,10 @@ +from __future__ import annotations + +from ._types import Tuple, Union, array + import numpy as np -def unique(x, /, *, return_counts=False, return_index=False, return_inverse=False, sorted=True): +def unique(x: array, /, *, return_counts: bool = False, return_index: bool = False, return_inverse: bool = False, sorted: bool = True) -> Union[array, Tuple[array, ...]]: """ Array API compatible wrapper for :py:func:`np.unique <numpy.unique>`. |