test: cancel test runs the real sampler at the end of the chain

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
digital-garbage
2026-09-27 19:10:23 +03:00
co-authored by Claude Opus 5.5
parent 3c187bda8a
commit 6acfd7e27b
+3 -1
View File
@@ -169,7 +169,9 @@ def test_alg_puts_the_sharp_anchor_back_when_a_run_is_cancelled():
class Executor:
class_obj = comfy.samplers.KSAMPLER(sampler_function)
def __call__(self, *a, **k):
raise AssertionError("should not fall back")
# The end of a real chain: the sampler's own sample(), running the
# sampler_function ALG swapped in for this call.
return self.class_obj.sample(*a, **k)
with pytest.raises(InterruptProcessingException):
captured["wrapper"](Executor(), Guider(), torch.tensor([0.9]), {}, None,