Acs 2

numpy-stubs/__init__.pyi

@@ -280,6 +280,7 @@ class ndarray(Generic[_DType]):
         self,
         key: Union[
             None,
+            range,
             slice,
             str,
             ndarray[_Int],
@@ -595,6 +596,9 @@ def asarray(a: ndarray, dtype: Type[str]) -> ndarray[str_]: ...
 def atleast_2d(ary: _Coercable) -> ndarray: ...
 @overload
 def atleast_2d(ar: _Coercable, *ary: _Coercable) -> List[ndarray]: ...
+def bincount(
+    x: _ArrayLike[_DType], weights: _ArrayLike = ..., minlength: int = ...
+) -> ndarray[_DType]: ...
 @overload
 def ceil(a: _FloatObj) -> _FloatObj: ...
 @overload
@@ -642,7 +646,7 @@ def isin(element: Sequence[_DType], test_element: _DType) -> ndarray[_DType]: ..
 @overload
 def isnan(x: float64) -> bool: ...
 @overload
-def isnan(x: ndarray[_DType]) -> ndarray[bool_]: ...
+def isnan(x: _ArrayLike[_DType]) -> ndarray[bool_]: ...
 @overload
 def ix_(x: ndarray[_DType]) -> ndarray[_DType]: ...
 @overload

numpy-stubs/random.pyi

@@ -1,6 +1,17 @@
 from typing import Iterable, List, Optional, Sequence, Tuple, TypeVar, Union, overload
 
-from . import _DType, _Float, _FloatObj, _Int, _IntObj, _ShapeType, float64, int64, ndarray
+from . import (
+    _DType,
+    _Float,
+    _FloatObj,
+    _Int,
+    _IntObj,
+    _ShapeType,
+    float64,
+    int64,
+    ndarray,
+    _ArrayLike,
+)
 
 _T = TypeVar("_T")
 @overload
@@ -43,6 +54,9 @@ def exponential(scale: _FloatObj, size: Sequence[_IntObj]) -> ndarray[float64]:
 def exponential(scale: Sequence[_FloatObj], size: Sequence[_IntObj]) -> ndarray[float64]: ...
 def geometric(p: float, size: _IntObj) -> ndarray[float64]: ...
 def get_state() -> Tuple[str, ndarray[_Int], _IntObj, _IntObj, _FloatObj]: ...
+def multinomial(
+    n: int, pvals: _ArrayLike[float] = ..., size: Union[int, Tuple[int, ...]] = ...
+) -> ndarray[_Int]: ...
 def normal(loc: float, scale: float, size: Union[int, Tuple[int, ...]]) -> ndarray[float64]: ...
 @overload
 def permutation(size: int) -> ndarray[int64]: ...