fix: remove redundant useless loop (#4656)

fix: potential error of undefined variable

https://github.com/comfyanonymous/ComfyUI/discussions/4650

script_examples/websockets_api_example.py

@@ -41,11 +41,10 @@ def get_images(ws, prompt):
             continue #previews are binary data
 
     history = get_history(prompt_id)[prompt_id]
-    for o in history['outputs']:
     for node_id in history['outputs']:
         node_output = history['outputs'][node_id]
-            if 'images' in node_output:
         images_output = []
+        if 'images' in node_output:
             for image in node_output['images']:
                 image_data = get_image(image['filename'], image['subfolder'], image['type'])
                 images_output.append(image_data)

tests/inference/test_execution.py

@@ -95,12 +95,11 @@ class ComfyClient:
                     pass # Probably want to store this off for testing
 
         history = self.get_history(prompt_id)[prompt_id]
-        for o in history['outputs']:
         for node_id in history['outputs']:
             node_output = history['outputs'][node_id]
             result.outputs[node_id] = node_output
-                if 'images' in node_output:
             images_output = []
+            if 'images' in node_output:
                 for image in node_output['images']:
                     image_data = self.get_image(image['filename'], image['subfolder'], image['type'])
                     image_obj = Image.open(BytesIO(image_data))

tests/inference/test_inference.py

@@ -109,11 +109,10 @@ class ComfyClient:
                 continue #previews are binary data
 
         history = self.get_history(prompt_id)[prompt_id]
-        for o in history['outputs']:
         for node_id in history['outputs']:
             node_output = history['outputs'][node_id]
-                if 'images' in node_output:
             images_output = []
+            if 'images' in node_output:
                 for image in node_output['images']:
                     image_data = self.get_image(image['filename'], image['subfolder'], image['type'])
                     images_output.append(image_data)