From 22c34c343a86bea7117a1dbc4cee440403b38025 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Sun, 21 May 2023 20:27:34 +1000 Subject: [PATCH] feat(nodes): fix types for InvocationServices --- invokeai/app/services/invocation_services.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/invokeai/app/services/invocation_services.py b/invokeai/app/services/invocation_services.py index 41654168d6..a85089554c 100644 --- a/invokeai/app/services/invocation_services.py +++ b/invokeai/app/services/invocation_services.py @@ -1,9 +1,9 @@ # Copyright (c) 2022 Kyle Schouviller (https://github.com/kyle0654) and the InvokeAI Team - +from typing import TYPE_CHECKING from logging import Logger + from invokeai.app.services.images import ImageService from invokeai.backend import ModelManager - from .events import EventServiceBase from .latent_storage import LatentsStorageBase from .image_file_storage import ImageFileStorageBase @@ -13,6 +13,11 @@ from .item_storage import ItemStorageABC from .config import InvokeAISettings +if TYPE_CHECKING: + from invokeai.app.services.graph import GraphExecutionState, LibraryGraph + from invokeai.app.services.invoker import InvocationProcessorABC + + class InvocationServices: """Services that can be used by invocations"""