From 242e0e6493f236cd1e7652c2c6f2f43c8b04f573 Mon Sep 17 00:00:00 2001 From: Jacob Segal Date: Tue, 8 Jul 2025 17:28:05 -0700 Subject: [PATCH] Resolve unit test in GPU-less runner --- tests-unit/websocket_feature_flags_test.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests-unit/websocket_feature_flags_test.py b/tests-unit/websocket_feature_flags_test.py index b8d7ee258..f79a3fd19 100644 --- a/tests-unit/websocket_feature_flags_test.py +++ b/tests-unit/websocket_feature_flags_test.py @@ -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