fix typing

getsentry · Sep 26, 2024 · 498987f · 498987f
1 parent 564ba1f
commit 498987f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/arroyo/processing/strategies/unfold.py b/arroyo/processing/strategies/unfold.py
@@ -1,9 +1,9 @@
 import time
 from collections import deque
-from typing import Callable, Iterable, Deque, Generic, Optional, TypeVar, Union, cast
+from typing import Callable, Deque, Generic, Iterable, Optional, TypeVar, Union, cast
 
 from arroyo.processing.strategies.abstract import MessageRejected, ProcessingStrategy
-from arroyo.types import FilteredPayload, Message
+from arroyo.types import BaseValue, FilteredPayload, Message
 
 TInput = TypeVar("TInput")
 TOutput = TypeVar("TOutput")
@@ -27,7 +27,7 @@ class Unfold(
 
     def __init__(
         self,
-        generator: Callable[[TInput], Iterable[TOutput]],
+        generator: Callable[[TInput], Iterable[BaseValue[TOutput]]],
         next_step: ProcessingStrategy[Union[FilteredPayload, TOutput]],
     ) -> None:
         self.__generator = generator