Skip to content

Commit 374cd9f

Browse files
committed
update unit test
Signed-off-by: Niels Bantilan <[email protected]>
1 parent e962d33 commit 374cd9f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Diff for: flytekit/bin/entrypoint.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,8 @@ def _dispatch_execute(
107107
if inspect.iscoroutine(outputs):
108108
# Handle eager-mode (async) tasks
109109
logger.info("Output is a coroutine")
110-
outputs = asyncio.get_event_loop().run_until_complete(outputs)
110+
with asyncio.Runner() as runner:
111+
outputs = runner.run(outputs)
111112

112113
# Step3a
113114
if isinstance(outputs, VoidPromise):

0 commit comments

Comments
 (0)