chore: Remove useless postfix

lablup · Aug 5, 2024 · 80babfd · 80babfd
1 parent 266fd30
commit 80babfd
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
@@ -31,9 +31,9 @@ def get_container_registry_cls(registry_info: ContainerRegistryRow) -> Type[Base
 
         cr_cls = HarborRegistry_v2
     elif registry_type == ContainerRegistryType.GITHUB:
-        from .github import GitHubRegistry_v2
+        from .github import GitHubRegistry
 
-        cr_cls = GitHubRegistry_v2
+        cr_cls = GitHubRegistry
     elif registry_type == ContainerRegistryType.LOCAL:
         from .local import LocalRegistry
 

diff --git a/src/ai/backend/manager/container_registry/github.py b/src/ai/backend/manager/container_registry/github.py
@@ -12,7 +12,7 @@
 log = BraceStyleAdapter(logging.getLogger(__spec__.name))  # type: ignore[name-defined]
 
 
-class GitHubRegistry_v2(BaseContainerRegistry):
+class GitHubRegistry(BaseContainerRegistry):
     async def fetch_repositories(
         self,
         sess: aiohttp.ClientSession,