From 6acfd7e27b4e00e85f52fc9d352fde030b7f6278 Mon Sep 17 00:00:00 2001 From: digital-garbage <133010426+digital-garbage@users.noreply.github.com> Date: Sun, 27 Sep 2026 19:10:23 +0300 Subject: [PATCH] test: cancel test runs the real sampler at the end of the chain Co-Authored-By: Claude Opus 5.5 --- modules/sampling/tests/test_cancel.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/modules/sampling/tests/test_cancel.py b/modules/sampling/tests/test_cancel.py index 4d83529f..c3972a96 100644 --- a/modules/sampling/tests/test_cancel.py +++ b/modules/sampling/tests/test_cancel.py @@ -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,