Fix task.status.status_str caused by #2666 (#4551)

* Fix task.status.status_str caused by 2666 regression * fix * fix
comfyanonymous · Aug 22, 2024 · bb4416d · bb4416d
1 parent c0b0da2
commit bb4416d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/execution.py b/execution.py
@@ -450,7 +450,7 @@ def handle_execution_error(self, prompt_id, prompt, current_outputs, executed, e
                 "current_outputs": list(current_outputs),
             }
             self.add_message("execution_error", mes, broadcast=False)
-        
+
     def execute(self, prompt, prompt_id, extra_data={}, execute_outputs=[]):
         nodes.interrupt_processing(False)
 
@@ -492,6 +492,7 @@ def execute(self, prompt, prompt_id, extra_data={}, execute_outputs=[]):
                     break
 
                 result, error, ex = execute(self.server, dynamic_prompt, self.caches, node_id, extra_data, executed, prompt_id, execution_list, pending_subgraph_results)
+                self.success = result != ExecutionResult.FAILURE
                 if result == ExecutionResult.FAILURE:
                     self.handle_execution_error(prompt_id, dynamic_prompt.original_prompt, current_outputs, executed, error, ex)
                     break