From ac6b0a1a7aaa48f02bc3759d5584ad9f924a39b6 Mon Sep 17 00:00:00 2001
From: Gyubong Lee <jopemachine@naver.com>
Date: Mon, 24 Jun 2024 04:23:32 +0000
Subject: [PATCH] fix: CI

---
 src/ai/backend/storage/migration.py | 3 ++-
 tests/storage-proxy/conftest.py     | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

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():