nodes/Limbicnation-ComfyUI-PromptGenerator
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

style: fix ruff formatting in test_prompt_refiner.py

Browse Source
This commit is contained in:
limbicnation
2026-05-01 02:42:32 +02:00
parent 662afb4286
commit ff26170a59
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/test_prompt_refiner.py
+12 -6
View File
@@ -17,9 +17,12 @@ class TestPromptRefinerSeed:
captured_seeds.append(seed)
return f"refined with seed={seed}"
with patch.object(node, "REFINEMENT_PROMPT", "{prompt}"), patch(
"nodes.prompt_refiner_node.OllamaClient.generate_streaming",
side_effect=_capture_seed,
with (
patch.object(node, "REFINEMENT_PROMPT", "{prompt}"),
patch(
"nodes.prompt_refiner_node.OllamaClient.generate_streaming",
side_effect=_capture_seed,
),
):
result = node.refine(
prompt="a forest",
@@ -43,9 +46,12 @@ class TestPromptRefinerSeed:
captured_seeds.append(seed)
return "refined"
with patch.object(node, "REFINEMENT_PROMPT", "{prompt}"), patch(
"nodes.prompt_refiner_node.OllamaClient.generate_streaming",
side_effect=_capture_seed,
with (
patch.object(node, "REFINEMENT_PROMPT", "{prompt}"),
patch(
"nodes.prompt_refiner_node.OllamaClient.generate_streaming",
side_effect=_capture_seed,
),
):
node.refine(
prompt="a forest",