From b8712f2494332796e18162386ab0daef03ddb2e2 Mon Sep 17 00:00:00 2001 From: bymyself Date: Sun, 1 Jun 2025 00:32:16 -0700 Subject: [PATCH] [fix] Improve CI compatibility by mocking nodes module import Prevent CUDA initialization chain by mocking the nodes module at import time, which is cleaner than deep mocking of CUDA-specific functions. --- tests-unit/comfy_extras_test/image_stitch_test.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/tests-unit/comfy_extras_test/image_stitch_test.py b/tests-unit/comfy_extras_test/image_stitch_test.py index 75c10aa64..7b7be0fdd 100644 --- a/tests-unit/comfy_extras_test/image_stitch_test.py +++ b/tests-unit/comfy_extras_test/image_stitch_test.py @@ -1,15 +1,17 @@ import torch import sys import os -from unittest.mock import patch +from unittest.mock import patch, MagicMock # Add the project root to Python path sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../..')) -# Mock CUDA-dependent modules to avoid CI failures -with patch('comfy.model_management.get_torch_device', return_value=torch.device('cpu')): - with patch('comfy.model_management.get_total_memory', return_value=8192): - from comfy_extras.nodes_images import ImageStitch +# Mock nodes module to prevent CUDA initialization during import +mock_nodes = MagicMock() +mock_nodes.MAX_RESOLUTION = 16384 + +with patch.dict('sys.modules', {'nodes': mock_nodes}): + from comfy_extras.nodes_images import ImageStitch class TestImageStitch: