diff --git a/src/ai/backend/storage/migration.py b/src/ai/backend/storage/migration.py index feee14c332..bd7d43d80f 100644 --- a/src/ai/backend/storage/migration.py +++ b/src/ai/backend/storage/migration.py @@ -25,6 +25,7 @@ EventProducer, ) from ai.backend.common.logging import BraceStyleAdapter, Logger +from ai.backend.common.types import LogSeverity from .abc import CAP_FAST_SIZE, AbstractVolume from .config import load_local_config, load_shared_config @@ -324,7 +325,7 @@ def main( Print migration script to OUTFILE. Pass - as OUTFILE to print results to STDOUT. """ - local_config = load_local_config(config_path, debug=debug) + local_config = load_local_config(config_path, LogSeverity.DEBUG, debug=debug) ipc_base_path = local_config["storage-proxy"]["ipc-base-path"] log_sockpath = Path( ipc_base_path / f"storage-proxy-logger-{os.getpid()}.sock", diff --git a/tests/storage-proxy/conftest.py b/tests/storage-proxy/conftest.py index a8acae6f52..4b7735587e 100644 --- a/tests/storage-proxy/conftest.py +++ b/tests/storage-proxy/conftest.py @@ -10,7 +10,7 @@ from ai.backend.common.etcd import AsyncEtcd, ConfigScopes from ai.backend.common.exception import ConfigurationError -from ai.backend.common.types import HostPortPair, QuotaScopeID, QuotaScopeType +from ai.backend.common.types import HostPortPair, LogSeverity, QuotaScopeID, QuotaScopeType from ai.backend.storage.abc import AbstractVolume from ai.backend.storage.config import load_local_config from ai.backend.storage.types import VFolderID @@ -47,7 +47,7 @@ def mock_etcd() -> Iterator[AsyncEtcd]: def has_backend(backend_name: str) -> dict[str, Any] | None: try: - local_config = load_local_config(None, debug=True) + local_config = load_local_config(None, LogSeverity.DEBUG, debug=True) except ConfigurationError: return None for _, info in local_config["volume"].items():