Make NestedList admit sequences.

numpy will actually accept not just lists, but also tuples as array-like.
wearepal · Mar 13, 2020 · 378db00 · 378db00
1 parent 3e37ef2
commit 378db00
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/numpy-stubs/__init__.pyi b/numpy-stubs/__init__.pyi
@@ -68,7 +68,13 @@ _FloatObj = TypeVar("_FloatObj", bound=Union[floating, float])
 _Int = TypeVar("_Int", bool_, int8, int16, int32, int64)
 _IntObj = TypeVar("_IntObj", bound=Union[integer, int])
 
-_NestedList = Union[List[_T], List[List[_T]], List[List[List[_T]]], List[List[List[List[_T]]]]]
+_NLT = TypeVar("_NLT", covariant=True)
+_NestedList = Union[
+    Sequence[_NLT],
+    Sequence[Sequence[_NLT]],
+    Sequence[Sequence[Sequence[_NLT]]],
+    Sequence[Sequence[Sequence[Sequence[_NLT]]]],
+]
 
 class ndarray(Generic[_DType]):
     """