How to set a different task runner for dev vs. prod (e.g. to use Dask only in prod)?

Set the environment variable on your infrastructure block (any infrastructure block supports adding
custom environment variables).

Make sure to specify the task_runner in the flow decorator, as shown here:

@flow(
    task_runner=DaskTaskRunner()
    if os.environ.get("ENVIRONMENT") == "prod"
    else ConcurrentTaskRunner(),
)
def my_flow():