8000 ENH: Make numpy.array_api more portable by asmeurer · Pull Request #25370 · numpy/numpy · GitHub
[go: up one dir, main page]
Skip to content

ENH: Make numpy.array_api more portable #25370

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 12 commits into from
Jan 21, 2024
Merged

ENH: Make numpy.array_api more portable #25370

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
3b20ad9
Replace device="cpu" with a special object in numpy.array_api
asmeurer Dec 11, 2023
13ab654
Use separate wrapped dtype objects in numpy.array_api
asmeurer Dec 11, 2023
28a15d5
Disallow 0-dimensional arrays in numpy.array_api.nonzero
asmeurer Dec 11, 2023
84a4605
Update some items in the array_api compatibility document
asmeurer Dec 12, 2023
64c7478
Add a release notes entry
asmeurer Dec 12, 2023
700ded8
Fix some type hint aliases for numpy.array_api
asmeurer Dec 13, 2023
4f2b67a
Fix bug in numpy.array_api.asarray with copy=True
asmeurer Jan 16, 2024
595342c
Make numpy.array_api dtypes issue a warning when compared against num…
asmeurer Jan 16, 2024
8e2af02
Fix some linter issues
asmeurer Jan 16, 2024
0915030
Add stacklevel to a warning
asmeurer Jan 16, 2024
61f97f0
Merge branch 'main' into array_api-portability
asmeurer Jan 16, 2024
7f354e5
Fix the array_api fft creation functions to use the custom CPU_DEVICE…
asmeurer Jan 17, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix some type hint aliases for numpy.array_api
  • Loading branch information
@asmeurer
asmeurer committed Dec 13, 2023
commit 700ded88944ed876e7453159d8441c14516727d5
31 changes: 4 additions & 27 deletions numpy/array_api/_typing.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,41 +29,18 @@
Protocol,
)

from ._array_object import Array, CPU_DEVICE
from numpy import (
dtype,
int8,
int16,
int32,
int64,
uint8,
uint16,
uint32,
uint64,
float32,
float64,
)
from ._array_object import Array, _cpu_device
from ._dtypes import _DType

_T_co = TypeVar("_T_co", covariant=True)

class NestedSequence(Protocol[_T_co]):
def __getitem__(self, key: int, /) -> _T_co | NestedSequence[_T_co]: ...
def __len__(self, /) -> int: ...

Device = type(CPU_DEVICE)
Device = _cpu_device

Dtype = dtype[Union[
int8,
int16,
int32,
int64,
uint8,
uint16,
uint32,
uint64,
float32,
float64,
]]
Dtype =_DType

if sys.version_info >= (3, 12):
from collections.abc import Buffer as SupportsBufferProtocol
Expand Down
0