diff --git a/horde_worker_regen/process_management/process_manager.py b/horde_worker_regen/process_management/process_manager.py index 94bd4ffe..a1aaa60f 100644 --- a/horde_worker_regen/process_management/process_manager.py +++ b/horde_worker_regen/process_management/process_manager.py @@ -7,7 +7,7 @@ import random import sys import time -from asyncio import CancelledError, Task +from asyncio import CancelledError from asyncio import Lock as Lock_Asyncio from collections import deque from collections.abc import Mapping @@ -872,7 +872,9 @@ def _end_inference_process(self, process_info: HordeProcessInfo) -> None: :return: None """ self._process_map.update_entry( - process_id=process_info.process_id, last_process_state=None, loaded_horde_model_name=None + process_id=process_info.process_id, + last_process_state=None, + loaded_horde_model_name=None, ) try: process_info.pipe_connection.send(HordeControlMessage(control_flag=HordeControlFlag.END_PROCESS))