github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

tests: fix broken tests

Browse Source
This commit is contained in:
psychedelicious
2024-02-08 00:36:53 +11:00
parent aff44c0e58
commit 6d25789705
4 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/aa_nodes/test_graph_execution_state.py
View File

@@ -60,7 +60,6 @@ def mock_services() -> InvocationServices:
image_records=None, # type: ignore
images=None, # type: ignore
invocation_cache=MemoryInvocationCache(max_cache_size=0),
latents=None, # type: ignore
logger=logging, # type: ignore
model_manager=None, # type: ignore
model_records=None, # type: ignore
@@ -74,6 +73,8 @@ def mock_services() -> InvocationServices:
session_queue=None, # type: ignore
urls=None, # type: ignore
workflow_records=None, # type: ignore
tensors=None,
conditioning=None,
)
@@ -89,7 +90,7 @@ def invoke_next(g: GraphExecutionState, services: InvocationServices) -> tuple[B
config=None,
context_data=None,
images=None,
latents=None,
tensors=None,
logger=None,
models=None,
util=None,

3
tests/aa_nodes/test_invoker.py
View File

@@ -63,7 +63,6 @@ def mock_services() -> InvocationServices:
image_records=None, # type: ignore
images=None, # type: ignore
invocation_cache=MemoryInvocationCache(max_cache_size=0),
latents=None, # type: ignore
logger=logging, # type: ignore
model_manager=None, # type: ignore
model_records=None, # type: ignore
@@ -77,6 +76,8 @@ def mock_services() -> InvocationServices:
session_queue=None, # type: ignore
urls=None, # type: ignore
workflow_records=None, # type: ignore
tensors=None,
conditioning=None,
)