fix: filter out none coroutines (#3060)

Jabolol · web-flow · commit a5ce7b5655e9 · 2025-02-19T16:42:22.000Z
diff --git a/warehouse/oso_dagster/utils/dlt.py b/warehouse/oso_dagster/utils/dlt.py
@@ -83,7 +83,9 @@ async def _wrapper(
             log = context.log if context else logger
 
             tasks: List[Coroutine[Any, Any, R]] = [
-                task() for task in fn(*args, **kwargs)
+                coro
+                for coro in (task() for task in fn(*args, **kwargs))
+                if coro is not None
             ]
             batches_list = list(batched(tasks, config.chunk_size))
 

Original file line number	Diff line number	Diff line change
`@@ -83,7 +83,9 @@ async def _wrapper(`
`83`	`83`	`log = context.log if context else logger`
`84`	`84`
`85`	`85`	`tasks: List[Coroutine[Any, Any, R]] = [`
`86`		`- task() for task in fn(args, *kwargs)`
	`86`	`+ coro`
	`87`	`+ for coro in (task() for task in fn(args, *kwargs))`
	`88`	`+ if coro is not None`
`87`	`89`	`]`
`88`	`90`	`batches_list = list(batched(tasks, config.chunk_size))`
`89`	`91`