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

fix(tests): remove graph library from test fixtures

Browse Source
This commit is contained in:
psychedelicious 2023-12-22 23:58:59 +11:00
parent 78b29db458
commit 756cb9c27e
2 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/aa_nodes/test_graph_execution_state.py
View File

@ -26,7 +26,6 @@ from invokeai.app.services.shared.graph import (
Graph, Graph,
GraphExecutionState, GraphExecutionState,
IterateInvocation, IterateInvocation,
LibraryGraph,
) )
from invokeai.backend.util.logging import InvokeAILogger from invokeai.backend.util.logging import InvokeAILogger
from tests.fixtures.sqlite_database import create_mock_sqlite_database from tests.fixtures.sqlite_database import create_mock_sqlite_database
@ -61,7 +60,6 @@ def mock_services() -> InvocationServices:
configuration=configuration, configuration=configuration,
events=TestEventService(), events=TestEventService(),
graph_execution_manager=graph_execution_manager, graph_execution_manager=graph_execution_manager,
graph_library=SqliteItemStorage[LibraryGraph](db=db, table_name="graphs"),
image_files=None, # type: ignore image_files=None, # type: ignore
image_records=None, # type: ignore image_records=None, # type: ignore
images=None, # type: ignore images=None, # type: ignore

3
tests/aa_nodes/test_invoker.py
View File

@ -24,7 +24,7 @@ from invokeai.app.services.invocation_stats.invocation_stats_default import Invo
from invokeai.app.services.invoker import Invoker from invokeai.app.services.invoker import Invoker
from invokeai.app.services.item_storage.item_storage_sqlite import SqliteItemStorage from invokeai.app.services.item_storage.item_storage_sqlite import SqliteItemStorage
from invokeai.app.services.session_queue.session_queue_common import DEFAULT_QUEUE_ID from invokeai.app.services.session_queue.session_queue_common import DEFAULT_QUEUE_ID
from invokeai.app.services.shared.graph import Graph, GraphExecutionState, GraphInvocation, LibraryGraph from invokeai.app.services.shared.graph import Graph, GraphExecutionState, GraphInvocation
@pytest.fixture @pytest.fixture
@ -66,7 +66,6 @@ def mock_services() -> InvocationServices:
configuration=configuration, configuration=configuration,
events=TestEventService(), events=TestEventService(),
graph_execution_manager=graph_execution_manager, graph_execution_manager=graph_execution_manager,
graph_library=SqliteItemStorage[LibraryGraph](db=db, table_name="graphs"),
image_files=None, # type: ignore image_files=None, # type: ignore
image_records=None, # type: ignore image_records=None, # type: ignore
images=None, # type: ignore images=None, # type: ignore