Add more logs

astronomer · Dec 19, 2024 · eb886e3 · eb886e3
1 parent 6be83ba
commit eb886e3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cosmos/listeners/dag_run_listener.py b/cosmos/listeners/dag_run_listener.py
@@ -68,6 +68,7 @@ def uses_cosmos(dag: DAG) -> bool:
 @hookimpl
 def on_dag_run_success(dag_run: DagRun, msg: str) -> None:
     logger.info("The on_dag_run_success was called")
+    print("The on_dag_run_success was called")
     dag = dag_run.get_dag()
     if not uses_cosmos(dag):
         return
@@ -88,6 +89,7 @@ def on_dag_run_success(dag_run: DagRun, msg: str) -> None:
 @hookimpl
 def on_dag_run_failed(dag_run: DagRun, msg: str) -> None:
     logger.info("The on_dag_run_failed was called")
+    print("The on_dag_run_failed was called")
     dag = dag_run.get_dag()
     if not uses_cosmos(dag):
         return