Resolve unit test in GPU-less runner

This commit is contained in:
Jacob Segal 2025-07-08 17:28:05 -07:00
parent bb483ab225
commit 242e0e6493

View File

@ -1,4 +1,7 @@
"""Simplified tests for WebSocket feature flags functionality."""
from comfy.cli_args import args
# Force CPU mode for unit testing
args.cpu = True
from unittest.mock import Mock, patch
from comfy_api import feature_flags