Is there a replacement for

upstream_tasks=[result])

in 2.0? I used this in 1.0 when using the Dask Executor to force a task to wait on another task, but I believe it was taken away in 2.0

it's now

wait_for