fix integration tests

src/backend/tests/unit/api/v1/test_transactions.py

@@ -128,8 +128,7 @@ def test_transform_transaction_table_single(self):
     def test_transform_transaction_table_list(self):
         """Test transforming a list of TransactionTable."""
         tables = [
-            TransactionTable(id=uuid4(), vertex_id=f"vertex-{i}", status="success", flow_id=uuid4())
-            for i in range(3)
+            TransactionTable(id=uuid4(), vertex_id=f"vertex-{i}", status="success", flow_id=uuid4()) for i in range(3)
         ]
 
         result = transform_transaction_table(tables)
@@ -152,8 +151,7 @@ def test_transform_transaction_table_for_logs_single(self):
     def test_transform_transaction_table_for_logs_list(self):
         """Test transforming a list of TransactionTable for logs view."""
         tables = [
-            TransactionTable(id=uuid4(), vertex_id=f"vertex-{i}", status="success", flow_id=uuid4())
-            for i in range(3)
+            TransactionTable(id=uuid4(), vertex_id=f"vertex-{i}", status="success", flow_id=uuid4()) for i in range(3)
         ]
 
         result = transform_transaction_table_for_logs(tables)

src/lfx/src/lfx/graph/vertex/base.py

@@ -105,7 +105,6 @@ def __init__(
         self.use_result = False
         self.build_times: list[float] = []
         self.state = VertexStates.ACTIVE
-        self.log_transaction_tasks: set[asyncio.Task] = set()
         self.output_names: list[str] = [
             output["name"] for output in self.outputs if isinstance(output, dict) and "name" in output
         ]
@@ -535,7 +534,7 @@ async def _log_transaction_async(
         target: Vertex | None = None,
         error=None,
     ) -> None:
-        """Log a transaction asynchronously with proper task handling and cancellation.
+        """Log a transaction asynchronously.
 
         Args:
             flow_id: The ID of the flow
@@ -544,15 +543,10 @@ async def _log_transaction_async(
             target: Optional target vertex
             error: Optional error information
         """
-        if self.log_transaction_tasks:
-            # Safely await and remove completed tasks
-            task = self.log_transaction_tasks.pop()
-            await task
-
-            # Create and track new task
-        task = asyncio.create_task(log_transaction(flow_id, source, status, target, error))
-        self.log_transaction_tasks.add(task)
-        task.add_done_callback(self.log_transaction_tasks.discard)
+        try:
+            await log_transaction(flow_id, source, status, target, error)
+        except Exception as exc:  # noqa: BLE001
+            logger.debug(f"Error logging transaction: {exc!s}")
 
     async def _get_result(
         self,