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

prune worker nodes downstream of branch collector

Browse Source
This commit is contained in:
Robert Wojciechowski
2026-03-02 07:54:44 +00:00
parent 0e7690b3a8
commit 8241bb9585
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/orchestration/prompt_transform.py
+2 -1
View File
@@ -317,7 +317,8 @@ def prune_prompt_for_worker(prompt_obj):
if branch_ids:
connected.update(_find_downstream_nodes(prompt_obj, branch_ids))
if branch_collector_ids:
connected.update(_find_downstream_nodes(prompt_obj, branch_collector_ids))
downstream_of_collectors = _find_downstream_nodes(prompt_obj, branch_collector_ids)
connected -= (downstream_of_collectors - set(branch_collector_ids))
pruned_prompt = {}
for node_id in connected:
tests/test_prompt_transform.py
+2 -2
View File
@@ -302,11 +302,11 @@ class PrunePromptForWorkerTests(unittest.TestCase):
self.assertIn("5", result)
self.assertIn("6", result)
def test_branch_collector_anchor_keeps_collector_and_downstream(self):
def test_branch_collector_anchor_keeps_collector_but_prunes_downstream(self):
prompt = _branch_collector_prompt()
result = pt.prune_prompt_for_worker(prompt)
self.assertIn("5", result)
self.assertIn("6", result)
self.assertNotIn("6", result)
# ---------------------------------------------------------------------------