From 525dfb8ec851c6f2ed738d8ddffe027f6bd10549 Mon Sep 17 00:00:00 2001 From: nate nowack Date: Thu, 23 Jan 2025 11:44:45 -0600 Subject: [PATCH] No transaction in `TaskRunRecorder` (#16829) --- src/prefect/server/services/task_run_recorder.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/prefect/server/services/task_run_recorder.py b/src/prefect/server/services/task_run_recorder.py index 33ae2f86e9c1..78ef9c3a6003 100644 --- a/src/prefect/server/services/task_run_recorder.py +++ b/src/prefect/server/services/task_run_recorder.py @@ -161,12 +161,13 @@ async def record_task_run_event(event: ReceivedEvent) -> None: } db = provide_database_interface() - async with db.session_context(begin_transaction=True) as session: + async with db.session_context() as session: await _insert_task_run(session, task_run, task_run_attributes) await _insert_task_run_state(session, task_run) await _update_task_run_with_state( session, task_run, denormalized_state_attributes ) + await session.commit() logger.debug( "Recorded task run state change",