fix: wait_function

Signed-off-by: SdgJlbl <[email protected]>
Substra · Feb 19, 2024 · ce26250 · ce26250
1 parent 32e354c
commit ce26250
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/substra/sdk/client.py b/substra/sdk/client.py
@@ -1101,7 +1101,11 @@ def wait_function(
         asset_getter = self.get_function
         status_canceled = models.FunctionStatus.canceled.value
         status_failed = models.FunctionStatus.failed.value
-        statuses_stopped = (models.FunctionStatus.ready.value, models.FunctionStatus.canceled.value)
+        statuses_stopped = (
+            models.FunctionStatus.ready.value,
+            models.FunctionStatus.canceled.value,
+            models.FunctionStatus.failed.value,
+        )
         return self._wait(
             key=key,
             asset_getter=asset_getter,