From 21d3d022473548fc5edd3ff9e81268961808ec8c Mon Sep 17 00:00:00 2001 From: jorenham Date: Thu, 6 Feb 2025 02:05:26 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=8F=B7=EF=B8=8F=20stub=20`numpy.=5Fglobal?= =?UTF-8?q?s`?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .github/workflows/ci.yml | 2 ++ src/numpy-stubs/__init__.pyi | 15 +++------------ src/numpy-stubs/_globals.pyi | 15 +++++++++++++++ 3 files changed, 20 insertions(+), 12 deletions(-) create mode 100644 src/numpy-stubs/_globals.pyi diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index ef1a3e41..3a1d5309 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -55,6 +55,7 @@ jobs: src/numpy-stubs/testing src/numpy-stubs/typing src/numpy-stubs/__config__.pyi + src/numpy-stubs/_globals.pyi src/numpy-stubs/_pytesttester.pyi src/numpy-stubs/dtypes.pyi src/numpy-stubs/exceptions.pyi @@ -77,6 +78,7 @@ jobs: src/numpy-stubs/testing src/numpy-stubs/typing src/numpy-stubs/__config__.pyi + src/numpy-stubs/_globals.pyi src/numpy-stubs/_pytesttester.pyi src/numpy-stubs/dtypes.pyi src/numpy-stubs/exceptions.pyi diff --git a/src/numpy-stubs/__init__.pyi b/src/numpy-stubs/__init__.pyi index 3526bb4a..c5e54279 100644 --- a/src/numpy-stubs/__init__.pyi +++ b/src/numpy-stubs/__init__.pyi @@ -2,7 +2,6 @@ import abc import builtins import ctypes as ct import datetime as dt -import enum import sys from _typeshed import StrOrBytesPath, SupportsFlush, SupportsLenAndGetItem, SupportsWrite from collections.abc import Callable, Iterable, Iterator, Mapping, Sequence @@ -467,6 +466,7 @@ from . import ( typing as npt, version as version, ) +from ._globals import _CopyMode @runtime_checkable class _Buffer(Protocol): @@ -1779,7 +1779,7 @@ class dtype(Generic[_SCT_co]): @final class flatiter(Generic[_ArrayT_co]): - __hash__: ClassVar[None] + __hash__: ClassVar[None] = None # type: ignore[assignment] # pyright: ignore[reportIncompatibleMethodOverride] @property def base(self) -> _ArrayT_co: ... @property @@ -1791,11 +1791,7 @@ class flatiter(Generic[_ArrayT_co]): def __next__(self: flatiter[NDArray[_SCT]]) -> _SCT: ... def __len__(self) -> int: ... @overload - def __getitem__( - self: flatiter[NDArray[_SCT]], - key: int | integer | tuple[int | integer], - /, - ) -> _SCT: ... + def __getitem__(self: flatiter[NDArray[_SCT]], key: int | integer | tuple[int | integer], /) -> _SCT: ... @overload def __getitem__( self, @@ -5280,11 +5276,6 @@ bitwise_right_shift: Final = right_shift permute_dims: Final = transpose pow: Final = power -class _CopyMode(enum.Enum): - ALWAYS: L[True] - IF_NEEDED: L[False] - NEVER: L[2] - class errstate: def __init__( self, diff --git a/src/numpy-stubs/_globals.pyi b/src/numpy-stubs/_globals.pyi new file mode 100644 index 00000000..c6b17d68 --- /dev/null +++ b/src/numpy-stubs/_globals.pyi @@ -0,0 +1,15 @@ +__all__ = ["_CopyMode", "_NoValue"] + +import enum +from typing import Final, final + +@final +class _CopyMode(enum.Enum): + ALWAYS = True + IF_NEEDED = False + NEVER = 2 + +@final +class _NoValueType: ... + +_NoValue: Final[_NoValueType] = ...