From 498987f29be50f63609a4a1a41bb9cd0356080fa Mon Sep 17 00:00:00 2001
From: Markus Unterwaditzer <markus-tarpit+git@unterwaditzer.net>
Date: Thu, 26 Sep 2024 16:55:55 +0200
Subject: [PATCH] fix typing

---
 arroyo/processing/strategies/unfold.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/arroyo/processing/strategies/unfold.py b/arroyo/processing/strategies/unfold.py
index 1560fc3d..2bfdd78f 100644
--- 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