diff --git a/src/ai/backend/manager/container_registry/__init__.py b/src/ai/backend/manager/container_registry/__init__.py index 394237558c..d3f773fbe2 100644 --- a/src/ai/backend/manager/container_registry/__init__.py +++ b/src/ai/backend/manager/container_registry/__init__.py @@ -33,9 +33,9 @@ def get_container_registry_cls(registry_info: Mapping[str, Any]) -> Type[BaseCon cr_cls = GitHubRegistry elif registry_type == "gitlab": - from .gitlab import GitLabRegistry_v2 + from .gitlab import GitLabRegistry - cr_cls = GitLabRegistry_v2 + cr_cls = GitLabRegistry elif registry_type == "local": from .local import LocalRegistry diff --git a/src/ai/backend/manager/container_registry/gitlab.py b/src/ai/backend/manager/container_registry/gitlab.py index f8c434d457..37413b52cc 100644 --- a/src/ai/backend/manager/container_registry/gitlab.py +++ b/src/ai/backend/manager/container_registry/gitlab.py @@ -13,7 +13,7 @@ log = BraceStyleAdapter(logging.getLogger(__spec__.name)) # type: ignore[name-defined] -class GitLabRegistry_v2(BaseContainerRegistry): +class GitLabRegistry(BaseContainerRegistry): async def fetch_repositories(self, sess: aiohttp.ClientSession) -> AsyncIterator[str]: name, access_token, gitlab_project = ( self.registry_info["username"],