Use Union until mlrun#606 is merged

gtopper · gtopper · commit e081b1a75e35 · 2026-01-22T19:24:02.000+07:00
diff --git a/storey/flow.py b/storey/flow.py
@@ -564,7 +564,7 @@ def _validate_not_already_streaming(self, event):
                 f"Step '{self.name}' received a streaming event from '{streaming_step}'."
             )
 
-    async def _emit_streaming_chunks(self, event, generator: Generator | AsyncGenerator) -> None:
+    async def _emit_streaming_chunks(self, event, generator: Union[Generator, AsyncGenerator]) -> None:
         """Emit streaming chunks from a generator, then send StreamCompletion.
 
         Args:

Original file line number	Diff line number	Diff line change
`@@ -564,7 +564,7 @@ def _validate_not_already_streaming(self, event):`
`564`	`564`	`f"Step '{self.name}' received a streaming event from '{streaming_step}'."`
`565`	`565`	`)`
`566`	`566`
`567`		`- async def _emit_streaming_chunks(self, event, generator: Generator \| AsyncGenerator) -> None:`
	`567`	`+ async def _emit_streaming_chunks(self, event, generator: Union[Generator, AsyncGenerator]) -> None:`
`568`	`568`	`"""Emit streaming chunks from a generator, then send StreamCompletion.`
`569`	`569`
`570`	`570`	`Args:`