fix: CI

lablup · Jun 24, 2024 · ac6b0a1 · ac6b0a1
1 parent 1e238c3
commit ac6b0a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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():