diff options
Diffstat (limited to 'tools/cru-py/cru/_iter.py')
| -rw-r--r-- | tools/cru-py/cru/_iter.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tools/cru-py/cru/_iter.py b/tools/cru-py/cru/_iter.py index 12d1d1f..5d3766a 100644 --- a/tools/cru-py/cru/_iter.py +++ b/tools/cru-py/cru/_iter.py @@ -415,8 +415,8 @@ class CruIterator(Generic[_T]): return self.first_n(1).single_or(fallback) @_wrap - def flatten(self) -> Iterable[_T | Iterable[_T]]: - return _Generic.iterable_flatten(self) + def flatten(self, max_depth: int = -1) -> Iterable[Any]: + return _Generic.iterable_flatten(self, max_depth) def select_by_indices(self, indices: Iterable[int]) -> CruIterator[_T]: index_set = set(indices) |