fixup! Generate random typestubs

Author: alanhdu

numpy-stubs/random/__init__.pyi

@@ -1,6 +1,14 @@
 import builtins
-from typing import Any, overload, Optional, Sequence, Tuple, TypeVar, Union
-
+from typing import (
+    Any,
+    List,
+    overload,
+    Optional,
+    Sequence,
+    Tuple,
+    TypeVar,
+    Union,
+)
 from numpy import ndarray
 
 _Size = Union[int, Sequence[int]]
@@ -444,7 +452,7 @@ def rayleigh(scale: float, size: None = ...) -> float: ...
 def rayleigh(scale: _ScalarOrArray[float], size: _Size) -> ndarray: ...
 @overload
 def rayleigh(scale: _ArrayLike[float], size: None = ...) -> ndarray: ...
-def seed(seed: Union[None, int, ndarray] = ...) -> None: ...
+def seed(seed: Union[None, int, Tuple[int], List[int]] = ...) -> None: ...
 def set_state(state: Tuple[str, ndarray, int, int, float]) -> None: ...
 def shuffle(x: _ArrayLike[Any]) -> None: ...
 @overload

numpy-stubs/random/mtrand.pyi

@@ -1,6 +1,14 @@
 import builtins
-from typing import Any, overload, Optional, Sequence, Tuple, TypeVar, Union
-
+from typing import (
+    Any,
+    List,
+    overload,
+    Optional,
+    Sequence,
+    Tuple,
+    TypeVar,
+    Union,
+)
 from numpy import ndarray
 
 _Size = Union[int, Sequence[int]]
@@ -9,7 +17,9 @@ _ArrayLike = Union[Sequence[_T], ndarray]
 _ScalarOrArray = Union[_T, Sequence[_T], ndarray]
 
 class RandomState:
-    def __init__(self, state: Optional[Union[int, ndarray]] = ...) -> None: ...
+    def __init__(
+        self, state: Union[None, int, List[int], Tuple[int]] = ...
+    ) -> None: ...
     @overload
     def beta(self, a: float, b: float, size: None = ...) -> float: ...
     @overload
@@ -544,7 +554,9 @@ class RandomState:
     def rayleigh(
         self, scale: _ArrayLike[float], size: None = ...
     ) -> ndarray: ...
-    def seed(self, seed: Union[None, int, ndarray] = ...) -> None: ...
+    def seed(
+        self, seed: Union[None, int, Tuple[int], List[int]] = ...
+    ) -> None: ...
     def set_state(
         self, state: Tuple[str, ndarray, int, int, float]
     ) -> None: ...