https://github.com/dagster-io/dagster/blob/9f220f20f3e440123216168cd913fa637cf94819/python_modules/dagster/dagster/core/execution/plan/execute_step.py#L473 has the relevant code but it's private