Return / raise even if complete

PrefectHQ · May 15, 2024 · e0efa72 · e0efa72
1 parent de4efbf
commit e0efa72
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/controlflow/core/task.py b/src/controlflow/core/task.py
@@ -302,10 +302,10 @@ def run(self, max_iterations: int = NOTSET) -> T:
                 )
             self.run_once()
             counter += 1
-            if self.is_successful():
-                return self.result
-            elif self.is_failed():
-                raise ValueError(f"{self.friendly_name()} failed: {self.error}")
+        if self.is_successful():
+            return self.result
+        elif self.is_failed():
+            raise ValueError(f"{self.friendly_name()} failed: {self.error}")
 
     @contextmanager
     def _context(self):