wehub-resource-sync
cddb07a176
build container image / cpu (push) Waiting to run
build container image / cuda (push) Waiting to run
build container image / rocm (push) Waiting to run
docs / deploy (push) Blocked by required conditions
docs / changes (push) Waiting to run
docs / check-and-build (push) Blocked by required conditions
frontend tests / frontend-tests (push) Waiting to run
openapi checks / openapi-checks (push) Waiting to run
python tests / py3.12: macos-default (push) Waiting to run
python tests / py3.11: windows-cpu (push) Waiting to run
python tests / py3.12: windows-cpu (push) Waiting to run
python tests / py3.11: linux-cpu (push) Waiting to run
python tests / py3.12: linux-cpu (push) Waiting to run
typegen checks / typegen-checks (push) Waiting to run
uv lock checks / uv-lock-checks (push) Waiting to run
frontend checks / frontend-checks (push) Waiting to run
lfs checks / lfs-check (push) Waiting to run
python checks / python-checks (push) Waiting to run
python tests / py3.11: macos-default (push) Waiting to run
36 lines
1.7 KiB
Python
36 lines
1.7 KiB
Python
import importlib
|
|
import pkgutil
|
|
|
|
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}"
|