diff --git a/robot-server/robot_server/runs/run_store.py b/robot-server/robot_server/runs/run_store.py index e32b962e6f3..6e9c735856f 100644 --- a/robot-server/robot_server/runs/run_store.py +++ b/robot-server/robot_server/runs/run_store.py @@ -101,7 +101,7 @@ def update_run_state( ) insert_command = sqlalchemy.insert(run_command_table) - select_run_resource = sqlalchemy.select(_run_columns).where( + select_run_resource = sqlalchemy.select(*_run_columns).where( run_table.c.id == run_id ) select_actions = sqlalchemy.select(action_table).where( @@ -216,7 +216,7 @@ def get(self, run_id: str) -> RunResource: Raises: RunNotFoundError: The given run ID was not found. """ - select_run_resource = sqlalchemy.select(_run_columns).where( + select_run_resource = sqlalchemy.select(*_run_columns).where( run_table.c.id == run_id ) @@ -240,7 +240,7 @@ def get_all(self, length: Optional[int] = None) -> List[RunResource]: Returns: All stored run entries. """ - select_runs = sqlalchemy.select(_run_columns) + select_runs = sqlalchemy.select(*_run_columns) select_actions = sqlalchemy.select(action_table).order_by(sqlite_rowid.asc()) actions_by_run_id = defaultdict(list)