From 9ec3ff9a71ec5d0018b2f91d6e575fc308a0ca88 Mon Sep 17 00:00:00 2001 From: Gyubong Lee Date: Mon, 5 Aug 2024 00:51:42 +0000 Subject: [PATCH] chore: remove _v2 --- src/ai/backend/manager/container_registry/__init__.py | 4 ++-- src/ai/backend/manager/container_registry/gitlab.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) 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"],