Skip to content

Commit ce65c22

Browse files
committed
fix and logs
1 parent 6ecb5a1 commit ce65c22

File tree

1 file changed

+11
-6
lines changed
  • services/director-v2/src/simcore_service_director_v2/modules/comp_scheduler

1 file changed

+11
-6
lines changed

services/director-v2/src/simcore_service_director_v2/modules/comp_scheduler/_scheduler_base.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -258,11 +258,16 @@ async def _set_schedule_done(
258258
project_id: ProjectID,
259259
iteration: Iteration,
260260
) -> None:
261-
await CompRunsRepository.instance(self.db_engine).mark_as_processed(
262-
user_id=user_id,
263-
project_id=project_id,
264-
iteration=iteration,
265-
)
261+
with log_context(
262+
_logger,
263+
logging.DEBUG,
264+
msg=f"mark pipeline run for {iteration=} for {user_id=} and {project_id=} as processed",
265+
):
266+
await CompRunsRepository.instance(self.db_engine).mark_as_processed(
267+
user_id=user_id,
268+
project_id=project_id,
269+
iteration=iteration,
270+
)
266271

267272
async def _set_states_following_failed_to_aborted(
268273
self, project_id: ProjectID, dag: nx.DiGraph, run_id: PositiveInt
@@ -738,7 +743,7 @@ async def _schedule_tasks_to_stop(
738743
)
739744
)
740745
for task in tasks_instantly_stopeable:
741-
comp_tasks[f"{task}"].state = RunningState.ABORTED
746+
comp_tasks[f"{task.node_id}"].state = RunningState.ABORTED
742747
# stop any remaining running task, these are already submitted
743748
if tasks_to_stop := [
744749
t for t in comp_tasks.values() if t.state in PROCESSING_STATES

0 commit comments

Comments
 (0)