2026-06-29 14:18:57 -04:00
1 changed files with 7 additions and 6 deletions
@@ -760,10 +760,11 @@ def enqueue_gpu_backfill(task_name: str) -> int:
        sel = sa_select(
            ImageRecord.id, literal(task_name), literal("pending")
        ).where(~already)
-        result = session.execute(
-            insert(GpuJob).from_select(
-                ["image_record_id", "task", "status"], sel
-            )
-        )
+        # RETURNING + count: result.rowcount is unreliable for INSERT…SELECT.
+        rows = session.execute(
+            insert(GpuJob)
+            .from_select(["image_record_id", "task", "status"], sel)
+            .returning(GpuJob.id)
+        ).fetchall()
        session.commit()
-        return result.rowcount or 0
+        return len(rows)