unknown tasks are still to be processed as completed.. since they are…

… lost and failed
ITISFoundation · Dec 20, 2023 · b6e9915 · b6e9915
1 parent ba8d106
commit b6e9915
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/...ices/director-v2/src/simcore_service_director_v2/modules/comp_scheduler/base_scheduler.py b/...ices/director-v2/src/simcore_service_director_v2/modules/comp_scheduler/base_scheduler.py
@@ -532,18 +532,16 @@ async def _update_states_from_comp_backend(
                 run_metadata=pipeline_params.run_metadata,
             )
 
-        if sorted_tasks.completed:
+        if sorted_tasks.completed or sorted_tasks.potentially_lost:
             await self._process_completed_tasks(
                 user_id,
-                sorted_tasks.completed,
+                sorted_tasks.completed + sorted_tasks.potentially_lost,
                 iteration,
                 pipeline_params=pipeline_params,
             )
 
-        if sorted_tasks.waiting or sorted_tasks.potentially_lost:
-            await self._process_waiting_tasks(
-                sorted_tasks.waiting + sorted_tasks.potentially_lost
-            )
+        if sorted_tasks.waiting:
+            await self._process_waiting_tasks(sorted_tasks.waiting)
 
     @abstractmethod
     async def _start_tasks(

diff --git a/...ices/director-v2/src/simcore_service_director_v2/modules/comp_scheduler/dask_scheduler.py b/...ices/director-v2/src/simcore_service_director_v2/modules/comp_scheduler/dask_scheduler.py
@@ -235,7 +235,7 @@ async def _process_completed_tasks(
     async def _process_task_result(
         self,
         task: CompTaskAtDB,
-        result: Exception | TaskOutputData,
+        result: BaseException | TaskOutputData,
         run_metadata: RunMetadataDict,
         iteration: Iteration,
     ) -> None: