Attempt to work around RuntimeError: UNEXPECTED COMMIT - THIS WILL BR…

…EAK HA LOCKS
astronomer · Dec 20, 2024 · 47fb46c · 47fb46c
1 parent 41a9293
commit 47fb46c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cosmos/listeners/dag_run_listener.py b/cosmos/listeners/dag_run_listener.py
@@ -47,8 +47,12 @@ def uses_cosmos(dag: DAG) -> bool:
     return bool(is_cosmos_dag(dag) or total_cosmos_task_groups(dag) or total_cosmos_tasks(dag))
 
 
+from airflow.utils.session import provide_session
+
+
+@provide_session
 @hookimpl
-def on_dag_run_success(dag_run: DagRun, msg: str) -> None:
+def on_dag_run_success(dag_run: DagRun, msg: str, session=None) -> None:
 
     logger.info(f"dir: {dir(dag_run.dag)}")
     logger.info("The on_dag_run_success was called")