mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-10-12 02:16:43 +00:00
Valeri Che
c5f630ef52
* Defer networkx import for schema generation * Declare networkx runtime dependency
70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
import importlib
|
|
import pkgutil
|
|
import subprocess
|
|
import sys
|
|
import textwrap
|
|
|
|
import invokeai
|
|
|
|
KNOWN_IMPORT_ERRORS = {
|
|
"invokeai.backend.image_util.normal_bae.nets.submodules.efficientnet_repo.setup",
|
|
"invokeai.backend.image_util.normal_bae.nets.submodules.efficientnet_repo.validate",
|
|
"invokeai.backend.image_util.normal_bae.nets.submodules.efficientnet_repo.caffe2_benchmark",
|
|
"invokeai.backend.image_util.normal_bae.nets.submodules.efficientnet_repo.caffe2_validate",
|
|
"invokeai.backend.image_util.normal_bae.nets.submodules.efficientnet_repo.onnx_optimize",
|
|
"invokeai.backend.image_util.normal_bae.nets.submodules.efficientnet_repo.onnx_validate",
|
|
"invokeai.backend.image_util.normal_bae.nets.submodules.efficientnet_repo.onnx_to_caffe",
|
|
"invokeai.backend.image_util.normal_bae.nets.submodules.efficientnet_repo.geffnet",
|
|
"invokeai.backend.image_util.normal_bae.nets.submodules.efficientnet_repo.onnx_export",
|
|
"invokeai.backend.model_manager.load.model_cache.torch_module_autocast.custom_modules.custom_invoke_linear_nf4",
|
|
"invokeai.backend.model_manager.load.model_cache.torch_module_autocast.custom_modules.custom_invoke_linear_8_bit_lt",
|
|
"invokeai.backend.image_util.normal_bae.nets.submodules.efficientnet_repo.hubconf",
|
|
"invokeai.backend.quantization.bnb_nf4",
|
|
"invokeai.backend.quantization.bnb_llm_int8",
|
|
}
|
|
|
|
|
|
def test_invokeai_imports():
|
|
modules = {name for _, name, _ in pkgutil.walk_packages(invokeai.__path__, invokeai.__name__ + ".")}
|
|
failed_to_import = set()
|
|
|
|
for mod in modules:
|
|
try:
|
|
importlib.import_module(mod)
|
|
except Exception:
|
|
if mod not in KNOWN_IMPORT_ERRORS:
|
|
failed_to_import.add(mod)
|
|
|
|
assert not failed_to_import, f"Modules failed to import: {failed_to_import}"
|
|
|
|
|
|
def test_graph_module_import_does_not_require_networkx():
|
|
script = """
|
|
import builtins
|
|
import sys
|
|
|
|
real_import = builtins.__import__
|
|
|
|
def blocked_import(name, globals=None, locals=None, fromlist=(), level=0):
|
|
if name == "networkx" or name.startswith("networkx."):
|
|
raise ModuleNotFoundError("No module named 'networkx'")
|
|
return real_import(name, globals, locals, fromlist, level)
|
|
|
|
builtins.__import__ = blocked_import
|
|
import invokeai.app.services.shared.graph # noqa: F401
|
|
|
|
assert "networkx" not in sys.modules
|
|
print("LAZY_OK")
|
|
"""
|
|
|
|
result = subprocess.run(
|
|
[sys.executable, "-c", textwrap.dedent(script)],
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=300,
|
|
check=False,
|
|
)
|
|
|
|
assert result.returncode == 0, result.stderr
|
|
assert "LAZY_OK" in result.stdout
|