chore: remove _v2

lablup · Aug 5, 2024 · 9ec3ff9 · 9ec3ff9
1 parent 4f790d3
commit 9ec3ff9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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"],