Fix flaky test output order (#3045)

This commit is contained in:
Nuno Campos
2025-01-15 09:02:05 -08:00
committed by GitHub
@@ -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)