fixup! Add generate framework

Author: alanhdu

generate/random.py

@@ -227,7 +227,7 @@ def choice(return_type: str, ty: str, size: bool) -> Func:
 functions += list(distribution("rayleigh", "float", Arg("scale", "float")))
 functions += [
     Func(
-        "seed", "None", [SELF, Arg("seed", "Union[None, int, ndarray]", True)]
+        "seed", "None", [SELF, Arg("seed", "Union[None, int, Tuple[int], List[int]]", True)]
     ),
     Func(
         "set_state",
@@ -273,8 +273,9 @@ def choice(return_type: str, ty: str, size: bool) -> Func:
 
 prelude = """\
 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]]
@@ -289,7 +290,7 @@ def choice(return_type: str, ty: str, size: bool) -> Func:
         """
 class RandomState:
     def __init__(
-        self, state: Optional[Union[int, ndarray]] = ...
+        self, state: Union[None, int, List[int], Tuple[int]] = ...
     ) -> None: ...
 """
     )