Merge pull request #84 from airflow-laminar/tkp/hf

Hotfix to ensure unconfigure supervisor task is present
airflow-laminar · Feb 15, 2025 · 5bc9f42 · 5bc9f42
2 parents 5374fda + a4b286d
commit 5bc9f42
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/airflow_supervisor/airflow/local.py b/airflow_supervisor/airflow/local.py
@@ -94,6 +94,9 @@ def initialize_tasks(self):
                 )
         else:
             self._stop_programs = PythonOperator(task_id=f"{self._dag.dag_id}-stop-programs", python_callable=skip)
+            self._unconfigure_supervisor = PythonOperator(
+                task_id=f"{self._dag.dag_id}-unconfigure-supervisor", python_callable=skip
+            )
 
         self._restart_programs = self.get_step_operator("restart-programs")
         self._stop_supervisor = self.get_step_operator("stop-supervisor")