Merge pull request #48 from airflow-laminar/tkp/ssh

Bugfix, use dag id from dag instance
airflow-laminar · Dec 12, 2024 · e180b2d · e180b2d
2 parents 9a67e04 + a0d55a0
commit e180b2d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/airflow_supervisor/airflow/ssh.py b/airflow_supervisor/airflow/ssh.py
@@ -91,7 +91,11 @@ def get_step_operator(self, step: SupervisorTaskStep) -> Operator:
             "force-kill",
         ):
             return SSHOperator(
-                **{"task_id": f"{self.dag_id}-{step}", **self.get_base_operator_kwargs(), **self.get_step_kwargs(step)}
+                **{
+                    "task_id": f"{self.dag.dag_id}-{step}",
+                    **self.get_base_operator_kwargs(),
+                    **self.get_step_kwargs(step),
+                }
             )
         else:
             # can be done via XMLRPC API