diff options
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/__init__.pyi | 7 | ||||
| -rw-r--r-- | numpy/core/_type_aliases.pyi | 19 |
2 files changed, 24 insertions, 2 deletions
diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi index f70965c41..c62227e5e 100644 --- a/numpy/__init__.pyi +++ b/numpy/__init__.pyi @@ -141,6 +141,11 @@ from numpy.core._asarray import ( require as require, ) +from numpy.core._type_aliases import ( + sctypes as sctypes, + sctypeDict as sctypeDict, +) + from numpy.core.numeric import ( zeros_like as zeros_like, ones as ones, @@ -460,8 +465,6 @@ save: Any savetxt: Any savez: Any savez_compressed: Any -sctypeDict: Any -sctypes: Any select: Any set_printoptions: Any set_string_function: Any diff --git a/numpy/core/_type_aliases.pyi b/numpy/core/_type_aliases.pyi new file mode 100644 index 000000000..6a1099cd3 --- /dev/null +++ b/numpy/core/_type_aliases.pyi @@ -0,0 +1,19 @@ +import sys +from typing import Dict, Union, Type, List + +from numpy import generic, signedinteger, unsignedinteger, floating, complexfloating + +if sys.version_info >= (3, 8): + from typing import TypedDict +else: + from typing_extensions import TypedDict + +class _SCTypes(TypedDict): + int: List[Type[signedinteger]] + uint: List[Type[unsignedinteger]] + float: List[Type[floating]] + complex: List[Type[complexfloating]] + others: List[type] + +sctypeDict: Dict[Union[int, str], Type[generic]] +sctypes: _SCTypes |