This commit is contained in:
Richard Yu 2025-12-02 19:33:22 -08:00
parent b874f469ce
commit 380b6aec8e
2 changed files with 10 additions and 21 deletions

View File

@ -133,22 +133,16 @@ def get_outputs_summary(outputs):
for item in items:
count += 1
if preview_output is not None:
continue
if not is_previewable(media_type, item):
continue
enriched = {
**item,
'nodeId': node_id,
'mediaType': media_type
}
if item.get('type') == 'output':
preview_output = enriched
elif fallback_preview is None:
fallback_preview = enriched
if preview_output is None and is_previewable(media_type, item):
enriched = {
**item,
'nodeId': node_id,
'mediaType': media_type
}
if item.get('type') == 'output':
preview_output = enriched
elif fallback_preview is None:
fallback_preview = enriched
return count, preview_output or fallback_preview

View File

@ -1,12 +1,7 @@
"""Unit tests for comfy_execution/jobs.py"""
import pytest
from pytest import fixture
from unittest.mock import MagicMock
from comfy_execution.jobs import (
JobStatus,
PREVIEWABLE_MEDIA_TYPES,
THREE_D_EXTENSIONS,
is_previewable,
normalize_queue_item,
normalize_history_item,