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,18 +133,12 @@ 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
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:

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,