re-structure
This commit is contained in:
@@ -26,7 +26,7 @@ def _collect_modules(package_name: str) -> list[str]:
|
||||
|
||||
|
||||
SHARED_MODULES = _collect_modules("shared")
|
||||
INFERENCE_MODULES = _collect_modules("inference")
|
||||
BACKEND_MODULES = _collect_modules("backend")
|
||||
TRAINING_MODULES = _collect_modules("training")
|
||||
|
||||
|
||||
@@ -36,9 +36,9 @@ def test_shared_module_imports(module_name: str) -> None:
|
||||
importlib.import_module(module_name)
|
||||
|
||||
|
||||
@pytest.mark.parametrize("module_name", INFERENCE_MODULES)
|
||||
def test_inference_module_imports(module_name: str) -> None:
|
||||
"""Every module in the inference package should import without error."""
|
||||
@pytest.mark.parametrize("module_name", BACKEND_MODULES)
|
||||
def test_backend_module_imports(module_name: str) -> None:
|
||||
"""Every module in the backend package should import without error."""
|
||||
importlib.import_module(module_name)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user