Skip to content

Commit cdcb0d8

Browse files
review @pcrespov
1 parent 5f9b7c5 commit cdcb0d8

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

packages/postgres-database/src/simcore_postgres_database/models/comp_runs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@
9999
doc="the run uses on demand clusters",
100100
),
101101
sa.Column(
102-
"dag_adjacency_list", JSONB, doc="Adjancey list for the pipeline's graph"
102+
"dag_adjacency_list", JSONB, doc="Adjacency list for the pipeline's graph"
103103
),
104104
sa.UniqueConstraint("project_uuid", "user_id", "iteration"),
105105
sa.Index("ix_comp_runs_user_id", "user_id"),

packages/postgres-database/src/simcore_postgres_database/utils_comp_run_snapshot_tasks.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import sqlalchemy as sa
44
from pydantic import PositiveInt
5+
from sqlalchemy.engine.row import Row
56
from sqlalchemy.ext.asyncio import AsyncConnection, AsyncEngine
67

78
from .models.comp_run_snapshot_tasks import comp_run_snapshot_tasks
@@ -40,9 +41,9 @@ async def update_for_run_id_and_node_id(
4041
run_id: PositiveInt,
4142
node_id: str,
4243
data: dict[str, Any],
43-
):
44+
) -> Row:
4445
async with pass_or_acquire_connection(engine, connection=conn) as _conn:
45-
result = await _conn.stream(
46+
result = await _conn.execute(
4647
comp_run_snapshot_tasks.update()
4748
.values(
4849
**data,
@@ -54,7 +55,7 @@ async def update_for_run_id_and_node_id(
5455
)
5556
.returning(*COMP_RUN_SNAPSHOT_TASKS_DB_COLS)
5657
)
57-
row = await result.one_or_none()
58+
row = result.one_or_none()
5859
if row is None:
5960
msg = f"update for run_id={run_id} and node_id={node_id} did not return any row"
6061
raise ValueError(msg)

0 commit comments

Comments
 (0)