From 1e7e97713cd9aff632dc342d8bf6eb0004c242b7 Mon Sep 17 00:00:00 2001 From: silveroxides Date: Mon, 13 Jul 2026 19:17:54 +0200 Subject: [PATCH] test: restore CLI state after node import --- tests-unit/comfy_extras_test/test_qwen_visual_fusion.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests-unit/comfy_extras_test/test_qwen_visual_fusion.py b/tests-unit/comfy_extras_test/test_qwen_visual_fusion.py index a03739eba..0e8b70b6d 100644 --- a/tests-unit/comfy_extras_test/test_qwen_visual_fusion.py +++ b/tests-unit/comfy_extras_test/test_qwen_visual_fusion.py @@ -2,10 +2,14 @@ import torch from comfy.cli_args import args as cli_args +prior_cpu = cli_args.cpu if not torch.cuda.is_available(): cli_args.cpu = True -from comfy_extras.nodes_qwen import TextEncodeQwenImageEditFusion, _flatten_images, _fuse_conditionings, _spatial_fusion_mask, _visual_token_span +try: + from comfy_extras.nodes_qwen import TextEncodeQwenImageEditFusion, _flatten_images, _fuse_conditionings, _spatial_fusion_mask, _visual_token_span +finally: + cli_args.cpu = prior_cpu def _tokens(image_position=1, suffix=1):