diff options
| author | iantra <iantrasolari@gmail.com> | 2022-09-05 14:32:34 +0200 |
|---|---|---|
| committer | iantra <iantrasolari@gmail.com> | 2022-09-05 17:22:03 +0200 |
| commit | d35f772dbfa598868bb0a1155ee7bf8ca9183b59 (patch) | |
| tree | c485cdf39d3f87013821aa1f38320b38d41d3088 /numpy/_typing | |
| parent | a670bea06280a50458a08e2dd9f1512150e02526 (diff) | |
| download | numpy-d35f772dbfa598868bb0a1155ee7bf8ca9183b59.tar.gz | |
BUG: Introduce Unknown array type to deal with NDArray[Any] more gracefully.
Diffstat (limited to 'numpy/_typing')
| -rw-r--r-- | numpy/_typing/__init__.py | 1 | ||||
| -rw-r--r-- | numpy/_typing/_array_like.py | 13 |
2 files changed, 14 insertions, 0 deletions
diff --git a/numpy/_typing/__init__.py b/numpy/_typing/__init__.py index 37ed06871..76b806c7f 100644 --- a/numpy/_typing/__init__.py +++ b/numpy/_typing/__init__.py @@ -198,6 +198,7 @@ from ._array_like import ( _ArrayLikeVoid_co as _ArrayLikeVoid_co, _ArrayLikeStr_co as _ArrayLikeStr_co, _ArrayLikeBytes_co as _ArrayLikeBytes_co, + _ArrayLikeUnknown as _ArrayLikeUnknown, ) from ._generic_alias import ( NDArray as NDArray, diff --git a/numpy/_typing/_array_like.py b/numpy/_typing/_array_like.py index 02f264222..2e5684b0b 100644 --- a/numpy/_typing/_array_like.py +++ b/numpy/_typing/_array_like.py @@ -141,3 +141,16 @@ _ArrayLikeInt = _DualArrayLike[ "dtype[integer[Any]]", int, ] + +# Extra ArrayLike type so that pyright can deal with NDArray[Any] +# Used as the first overload, should only match NDArray[Any], +# not any actual types. +# https://github.com/numpy/numpy/pull/22193 +class _UnknownType: + ... + + +_ArrayLikeUnknown = _DualArrayLike[ + "dtype[_UnknownType]", + _UnknownType, +] |