Run ruff

Author: brandonrising

invokeai/app/services/session_processor/session_processor_base.py

@@ -5,6 +5,7 @@
 from invokeai.app.services.session_processor.session_processor_common import SessionProcessorStatus
 from invokeai.app.services.session_queue.session_queue_common import SessionQueueItem
 
+
 class SessionRunnerBase(ABC):
     """
     Base class for session runner.

invokeai/app/services/session_processor/session_processor_default.py

@@ -2,7 +2,7 @@
 from contextlib import suppress
 from threading import BoundedSemaphore, Thread
 from threading import Event as ThreadEvent
-from typing import Optional, Union, Callable
+from typing import Callable, Optional, Union
 
 from fastapi_events.handlers.local import local_handler
 from fastapi_events.typing import Event as FastAPIEvent
@@ -16,7 +16,7 @@
 from invokeai.app.util.profiler import Profiler
 
 from ..invoker import Invoker
-from .session_processor_base import SessionProcessorBase, SessionRunnerBase, InvocationServices
+from .session_processor_base import InvocationServices, SessionProcessorBase, SessionRunnerBase
 from .session_processor_common import SessionProcessorStatus
 
 
@@ -36,7 +36,9 @@ def start(self, services: InvocationServices, cancel_event: ThreadEvent):
         self.services = services
         self.cancel_event = cancel_event
 
-    def next_invocation(self, previous_invocation: Optional[BaseInvocation], queue_item: SessionQueueItem, cancel_event: ThreadEvent) -> Optional[BaseInvocation]:
+    def next_invocation(
+        self, previous_invocation: Optional[BaseInvocation], queue_item: SessionQueueItem, cancel_event: ThreadEvent
+    ) -> Optional[BaseInvocation]:
         invocation = None
         if not (queue_item.session.is_complete() or cancel_event.is_set()):
             try:
@@ -57,7 +59,9 @@ def next_invocation(self, previous_invocation: Optional[BaseInvocation], queue_i
                     queue_id=queue_item.queue_id,
                     graph_execution_state_id=queue_item.session.id,
                     node=previous_invocation.model_dump() if previous_invocation else {},
-                    source_node_id=queue_item.session.prepared_source_mapping[previous_invocation.id] if previous_invocation else "",
+                    source_node_id=queue_item.session.prepared_source_mapping[previous_invocation.id]
+                    if previous_invocation
+                    else "",
                     error_type=exc.__class__.__name__,
                     error=node_error,
                 )
@@ -67,7 +71,6 @@ def next_invocation(self, previous_invocation: Optional[BaseInvocation], queue_i
             invocation = None
         return invocation
 
-
     def run(self, queue_item: SessionQueueItem):
         """Run the graph"""
         if not queue_item.session:
@@ -192,12 +195,11 @@ def run_node(self, node_id: str, queue_item: SessionQueueItem):
             )
 
 
-
 class DefaultSessionProcessor(SessionProcessorBase):
-
     def __init__(self, session_runner: Union[SessionRunnerBase, None] = None) -> None:
         super().__init__()
         self.session_runner = session_runner if session_runner else DefaultSessionRunner()
+
     def start(
         self,
         invoker: Invoker,