From 63d2cb3e0c4242e686b4aa5617f98a760aefd1b4 Mon Sep 17 00:00:00 2001
From: asagi4 <130366179+asagi4@users.noreply.github.com>
Date: Tue, 20 May 2025 21:28:11 +0300
Subject: [PATCH] Tests are broken again...

---
 .github/workflows/tests_with_comfy.yml | 2 +-
 prompt_control/test_graph.py           | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/.github/workflows/tests_with_comfy.yml b/.github/workflows/tests_with_comfy.yml
index b660e63..0d89b28 100644
--- a/.github/workflows/tests_with_comfy.yml
+++ b/.github/workflows/tests_with_comfy.yml
@@ -23,5 +23,5 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: '3.11'
-      - run: pip install -r requirements.txt
+      - run: pip install -r requirements.txt -r ComfyUI/requirements.txt
       - run: PYTHONPATH=ComfyUI python -m prompt_control.test_graph
diff --git a/prompt_control/test_graph.py b/prompt_control/test_graph.py
index 01fc3de..8f7c45d 100644
--- a/prompt_control/test_graph.py
+++ b/prompt_control/test_graph.py
@@ -17,7 +17,6 @@ def apply(cls, text, **kwargs):
 
 
 @mock.patch("prompt_control.utils.lora_name_to_file", find_file)
-@mock.patch.dict("sys.modules", nodes=mock.MagicMock())
 class GraphTests(unittest.TestCase):
     maxDiff = 4096