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: for item in items:
count += 1 count += 1
if preview_output is not None: if preview_output is None and is_previewable(media_type, item):
continue
if not is_previewable(media_type, item):
continue
enriched = { enriched = {
**item, **item,
'nodeId': node_id, 'nodeId': node_id,
'mediaType': media_type 'mediaType': media_type
} }
if item.get('type') == 'output': if item.get('type') == 'output':
preview_output = enriched preview_output = enriched
elif fallback_preview is None: elif fallback_preview is None:

View File

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