Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
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
223 changes: 114 additions & 109 deletions src/numpy-stubs/__init__.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -32,154 +32,170 @@ from uuid import UUID
import numpy as np
from numpy.__config__ import show as show_config
from numpy._array_api_info import __array_namespace_info__
from numpy._core._asarray import require
from numpy._core._internal import _ctypes
from numpy._core._type_aliases import sctypeDict
from numpy._core._ufunc_config import _ErrCall, _ErrKind, getbufsize, geterr, geterrcall, setbufsize, seterr, seterrcall
from numpy._core.arrayprint import (
array2string,
array_repr,
array_str,
format_float_positional,
format_float_scientific,
get_printoptions,
printoptions,
set_printoptions,
)
from numpy._core.einsumfunc import einsum, einsum_path
from numpy._core.fromnumeric import (
from numpy._core import (
False_,
ScalarType,
True_,
all,
allclose,
amax,
amin,
any,
arange,
argmax,
argmin,
argpartition,
argsort,
argwhere,
around,
choose,
clip,
compress,
cumprod,
cumsum,
cumulative_prod,
cumulative_sum,
diagonal,
matrix_transpose,
max,
mean,
min,
ndim,
nonzero,
partition,
prod,
ptp,
put,
ravel,
repeat,
reshape,
resize,
round,
searchsorted,
shape,
size,
sort,
squeeze,
std,
sum,
swapaxes,
take,
trace,
transpose,
var,
)
from numpy._core.function_base import geomspace, linspace, logspace
from numpy._core.getlimits import finfo, iinfo
from numpy._core.memmap import memmap
from numpy._core.multiarray import (
arange,
array,
array2string,
array_equal,
array_equiv,
array_repr,
array_str,
asanyarray,
asarray,
ascontiguousarray,
asfortranarray,
bincount,
astype,
atleast_1d,
atleast_2d,
atleast_3d,
base_repr,
binary_repr,
block,
broadcast,
busday_count,
busday_offset,
busdaycalendar,
can_cast,
choose,
clip,
complex192,
complex256,
compress,
concatenate,
convolve,
copyto,
correlate,
count_nonzero,
cross,
cumprod,
cumsum,
cumulative_prod,
cumulative_sum,
datetime_as_string,
datetime_data,
diagonal,
dot,
einsum,
einsum_path,
empty,
empty_like,
flagsobj,
finfo,
flatiter,
flatnonzero,
float96,
float128,
format_float_positional,
format_float_scientific,
from_dlpack,
frombuffer,
fromfile,
fromfunction,
fromiter,
frompyfunc,
fromstring,
inner,
is_busday,
lexsort,
may_share_memory,
min_scalar_type,
nditer,
nested_iters,
packbits,
promote_types,
putmask,
result_type,
shares_memory,
unpackbits,
vdot,
where,
zeros,
)
from numpy._core.numeric import (
False_,
True_,
allclose,
argwhere,
array_equal,
array_equiv,
astype,
base_repr,
binary_repr,
convolve,
correlate,
count_nonzero,
cross,
flatnonzero,
fromfunction,
full,
full_like,
geomspace,
get_printoptions,
getbufsize,
geterr,
geterrcall,
hstack,
identity,
iinfo,
indices,
inf,
inner,
is_busday,
isclose,
isdtype,
isfortran,
isscalar,
issubdtype,
lexsort,
linspace,
little_endian,
logspace,
matrix_transpose,
max,
may_share_memory,
mean,
min,
min_scalar_type,
moveaxis,
nan,
ndim,
nditer,
nested_iters,
newaxis,
nonzero,
ones,
ones_like,
outer,
partition,
printoptions,
prod,
promote_types,
ptp,
put,
putmask,
ravel,
recarray,
record,
repeat,
require,
reshape,
resize,
result_type,
roll,
rollaxis,
round,
sctypeDict,
searchsorted,
set_printoptions,
setbufsize,
seterr,
seterrcall,
shape,
shares_memory,
size,
sort,
squeeze,
stack,
std,
sum,
swapaxes,
take,
tensordot,
trace,
transpose,
typecodes,
unstack,
var,
vdot,
vstack,
where,
zeros,
zeros_like,
)
from numpy._core.numerictypes import ScalarType, isdtype, issubdtype, typecodes
from numpy._core.records import recarray, record
from numpy._core.shape_base import atleast_1d, atleast_2d, atleast_3d, block, hstack, stack, unstack, vstack
from numpy._core._internal import _ctypes
from numpy._core._ufunc_config import _ErrCall, _ErrKind
from numpy._core.memmap import memmap
from numpy._core.multiarray import bincount, flagsobj, packbits, unpackbits
from numpy._pytesttester import PytestTester
from numpy._typing import (
ArrayLike,
Expand Down Expand Up @@ -302,20 +318,9 @@ from numpy._typing._callable import (
_UnsignedIntMod,
_UnsignedIntOp,
)
from numpy._typing._extended_precision import (
complex160,
complex192,
complex256,
complex512,
float80,
float96,
float128,
float256,
int128,
int256,
uint128,
uint256,
)

# TODO(jorenham): remove these: https://github.com/numpy/numtype/issues/120
from numpy._typing._extended_precision import complex160, complex512, float80, float256, int128, int256, uint128, uint256
from numpy.lib import scimath as emath
from numpy.lib._arraypad_impl import pad
from numpy.lib._arraysetops_impl import (
Expand Down
Loading
Loading