Fix flaky test output order

This commit is contained in:
Nuno Campos
2025-01-15 08:45:35 -08:00
parent 65a41942ef
commit b7e4656c91
@@ -2456,6 +2456,7 @@ async def test_imp_task(checkpointer_name: str) -> None:
async def mapper(input: int) -> str:
nonlocal mapper_calls
mapper_calls += 1
await asyncio.sleep(0.1 * input)
return str(input) * 2
@entrypoint(checkpointer=checkpointer)