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 pytest
import torch 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.install.model_install_backend import ModelInstall
from invokeai.backend.model_management.model_manager import ModelInfo from invokeai.backend.model_management.model_manager import ModelInfo
from invokeai.backend.model_management.models.base import BaseModelType, ModelNotFoundException, ModelType, SubModelType 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 # must parse config before importing Graph, so its nodes union uses the config
conf = InvokeAIAppConfig().get_config() conf = InvokeAIAppConfig().get_config()
conf.parse_args(conf=allow_deny_nodes_conf, argv=[]) 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 # confirm graph validation fails when using denied node
Graph(nodes={"1": {"id": "1", "type": "integer"}}) Graph(nodes={"1": {"id": "1", "type": "integer"}})