Revert "fix: fix test imports"

This reverts commit 9646157ad5.
This commit is contained in:
psychedelicious 2023-10-13 21:47:51 +11:00
parent 2716ae353b
commit fadfe1dfe9
No known key found for this signature in database
2 changed files with 2 additions and 2 deletions

View File

@ -5,7 +5,7 @@ from typing import Optional, Union
import pytest
import torch
from invokeai.app.services.config.config_default import InvokeAIAppConfig
from invokeai.app.services.config.invokeai_config import InvokeAIAppConfig
from invokeai.backend.install.model_install_backend import ModelInstall
from invokeai.backend.model_management.model_manager import ModelInfo
from invokeai.backend.model_management.models.base import BaseModelType, ModelNotFoundException, ModelType, SubModelType

View File

@ -192,7 +192,7 @@ def test_deny_nodes(patch_rootdir):
# must parse config before importing Graph, so its nodes union uses the config
conf = InvokeAIAppConfig().get_config()
conf.parse_args(conf=allow_deny_nodes_conf, argv=[])
from invokeai.app.services.shared.graph import Graph
from invokeai.app.services.graph import Graph
# confirm graph validation fails when using denied node
Graph(nodes={"1": {"id": "1", "type": "integer"}})