Merge pull request #53 from airflow-laminar/tkp/hf4

Fix for cfg attribute
airflow-laminar · Dec 17, 2024 · bede7c9 · bede7c9
2 parents ffa8b54 + 555fa4e
commit bede7c9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/airflow_supervisor/airflow/ssh.py b/airflow_supervisor/airflow/ssh.py
@@ -55,7 +55,7 @@ def get_step_kwargs(self, step: SupervisorTaskStep) -> Dict:
                 **self._ssh_operator_kwargs,
                 "command": f"""
 {self._command_prefix}
-_airflow_supervisor_command {step} '{self._supervisor_cfg.model_dump_json()}'
+_airflow_supervisor_command {step} '{self._cfg.model_dump_json()}'
 """,
             }
         elif step in ("start-supervisor", "stop-supervisor", "unconfigure-supervisor", "force-kill"):
@@ -64,7 +64,7 @@ def get_step_kwargs(self, step: SupervisorTaskStep) -> Dict:
                 **self._ssh_operator_kwargs,
                 "command": f"""
 {self._command_prefix}
-_airflow_supervisor_command {step} --cfg {quote(str(self._supervisor_cfg._pydantic_path))}
+_airflow_supervisor_command {step} --cfg {quote(str(self._cfg._pydantic_path))}
 """,
             }
         else: