Add debug messages

astronomer · Dec 19, 2024 · 6be83ba · 6be83ba
1 parent 7ef7c25
commit 6be83ba
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cosmos/listeners/dag_run_listener.py b/cosmos/listeners/dag_run_listener.py
@@ -67,8 +67,8 @@ 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")
     dag = dag_run.get_dag()
-
     if not uses_cosmos(dag):
         return
 
@@ -87,6 +87,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")
     dag = dag_run.get_dag()
     if not uses_cosmos(dag):
         return