diff --git a/airflow-core/src/airflow/ui/src/components/Clear/Run/ClearRunButton.tsx b/airflow-core/src/airflow/ui/src/components/Clear/Run/ClearRunButton.tsx index 0c396b7b21214..6cbc3aa08b004 100644 --- a/airflow-core/src/airflow/ui/src/components/Clear/Run/ClearRunButton.tsx +++ b/airflow-core/src/airflow/ui/src/components/Clear/Run/ClearRunButton.tsx @@ -55,7 +55,7 @@ const ClearRunButton = ({ dagRun, isHotkeyEnabled = false }: Props) => { > - {open ? : undefined} + ); }; diff --git a/airflow-core/src/airflow/ui/src/components/Clear/Run/ClearRunDialog.tsx b/airflow-core/src/airflow/ui/src/components/Clear/Run/ClearRunDialog.tsx index 05f15856c6bfe..74b0cf2fb3571 100644 --- a/airflow-core/src/airflow/ui/src/components/Clear/Run/ClearRunDialog.tsx +++ b/airflow-core/src/airflow/ui/src/components/Clear/Run/ClearRunDialog.tsx @@ -45,6 +45,14 @@ const ClearRunDialog = ({ dagRun, onClose, open }: Props) => { const [note, setNote] = useState(dagRun.note); const [selectedOptions, setSelectedOptions] = useState>(["existingTasks"]); + + const handleClose = () => { + setNote(dagRun.note); + setSelectedOptions(["existingTasks"]); + onClose(); + }; + + const onlyFailed = selectedOptions.includes("onlyFailed"); const onlyNew = selectedOptions.includes("newTasks"); @@ -62,6 +70,7 @@ const ClearRunDialog = ({ dagRun, onClose, open }: Props) => { dagId, dagRunId, options: { + enabled: open, refetchInterval: (query) => query.state.data?.task_instances.some((ti) => "state" in ti && isStatePending(ti.state)) ? refetchInterval @@ -77,13 +86,13 @@ const ClearRunDialog = ({ dagRun, onClose, open }: Props) => { const { isPending, mutate } = useClearDagRun({ dagId, dagRunId, - onSuccessConfirm: onClose, + onSuccessConfirm: handleClose, }); const { isPending: isPendingPatchDagRun, mutate: mutatePatchDagRun } = usePatchDagRun({ dagId, dagRunId, - onSuccess: onClose, + onSuccess: handleClose, }); // Check if DAG versions differ (works for both bundle-versioned and local bundles) @@ -96,7 +105,7 @@ const ClearRunDialog = ({ dagRun, onClose, open }: Props) => { const shouldShowBundleVersionOption = versionsDiffer && !onlyNew; return ( - +