Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

add restoration services to nodes

Browse Source
This commit is contained in:
Lincoln Stein
2023-03-11 17:00:00 -05:00
parent 3aa1ee1218
commit 8ca91b1774
9 changed files with 125 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/nodes/test_graph_execution_state.py
View File

@ -26,7 +26,8 @@ def mock_services():
images = None,
queue = MemoryInvocationQueue(),
graph_execution_manager = SqliteItemStorage[GraphExecutionState](filename = sqlite_memory, table_name = 'graph_executions'),
processor = DefaultInvocationProcessor()
processor = DefaultInvocationProcessor(),
restoration = None,
)
def invoke_next(g: GraphExecutionState, services: InvocationServices) -> tuple[BaseInvocation, BaseInvocationOutput]:

3
tests/nodes/test_invoker.py
View File

@ -26,7 +26,8 @@ def mock_services() -> InvocationServices:
images = None, # type: ignore
queue = MemoryInvocationQueue(),
graph_execution_manager = SqliteItemStorage[GraphExecutionState](filename = sqlite_memory, table_name = 'graph_executions'),
processor = DefaultInvocationProcessor()
processor = DefaultInvocationProcessor(),
restoration = None,
)
@pytest.fixture()