From e8ab752975de587fb146df271a9b10a396dc75d6 Mon Sep 17 00:00:00 2001 From: Gyubong Lee Date: Sat, 3 Aug 2024 05:29:32 +0000 Subject: [PATCH] chore: Remove useless postfix --- src/ai/backend/manager/container_registry/__init__.py | 4 ++-- src/ai/backend/manager/container_registry/github.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 31b6b3a651..7da3ebb3a6 100644 --- 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 index 900b6f7e0e..942253ee8e 100644 --- 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,