From 2fd5fc81f5bd0d8e3dd844cc5a1d437efc72e8c4 Mon Sep 17 00:00:00 2001 From: Benjamin Berman Date: Sat, 26 Jul 2025 17:24:46 -0700 Subject: [PATCH] This raises ProcessExpired on Linux but not on Windows --- tests/unit/test_panics.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/unit/test_panics.py b/tests/unit/test_panics.py index d7919c402..251ed9d97 100644 --- a/tests/unit/test_panics.py +++ b/tests/unit/test_panics.py @@ -4,6 +4,7 @@ from unittest.mock import patch import pytest import torch +import pebble.common.types from comfy.cli_args_types import Configuration from comfy.client.embedded_comfy_client import Comfy @@ -156,7 +157,7 @@ async def test_panic_on_exception_with_executor(executor_cls, executor_kwargs): async with Comfy(configuration=config, executor=executor) as client: # Queue our failing workflow await client.queue_prompt(create_failing_workflow()) - except SystemExit: + except (SystemExit, pebble.common.types.ProcessExpired): sys_exit_called = True except UnrecoverableError: # We expect the exception to be raised here