diff --git a/js_modules/dagit/src/runs/RunActionButtons.tsx b/js_modules/dagit/src/runs/RunActionButtons.tsx --- a/js_modules/dagit/src/runs/RunActionButtons.tsx +++ b/js_modules/dagit/src/runs/RunActionButtons.tsx @@ -73,17 +73,18 @@ disabled={inFlight} onClick={async () => { setInFlight(true); - const res = await cancel({ + const {data} = await cancel({ variables: {runId: run.runId}, }); - setInFlight(false); - if (res.data?.terminatePipelineExecution?.message) { + const message = data?.terminatePipelineExecution?.message; + if (message) { SharedToaster.show({ - message: res.data.terminatePipelineExecution.message, + message, icon: 'error', intent: Intent.DANGER, }); } + setInFlight(false); }} /> ); diff --git a/js_modules/dagit/src/runs/RunUtils.tsx b/js_modules/dagit/src/runs/RunUtils.tsx --- a/js_modules/dagit/src/runs/RunUtils.tsx +++ b/js_modules/dagit/src/runs/RunUtils.tsx @@ -257,6 +257,9 @@ canTerminate } } + ... on PythonError { + message + } } } `;