diff --git a/metaspace/engine/sm/engine/annotation_lithops/executor.py b/metaspace/engine/sm/engine/annotation_lithops/executor.py index 0df26a2f6..d24dd3c8d 100644 --- a/metaspace/engine/sm/engine/annotation_lithops/executor.py +++ b/metaspace/engine/sm/engine/annotation_lithops/executor.py @@ -265,7 +265,7 @@ def map( # failed futures failed_idxs = [i for i, f in enumerate(futures or []) if f.error] # pylint: disable=unsubscriptable-object # (because futures is Optional) - failed_activation_ids = [futures[i].activation_id for i in failed_idxs] + failed_activation_ids = [futures[i].activation_id for i in failed_idxs] if futures else [] self._perf.record_entry( func_name, @@ -275,7 +275,7 @@ def map( attempt=attempt, runtime_memory=runtime_memory, failed_activation_ids=failed_activation_ids, - request_ids=[f.activation_id for f in futures], # pylint: disable=not-an-iterable + request_ids=[f.activation_id for f in (futures or [])], # pylint: disable=not-an-iterable ) if (