fix: Support private repository login

lablup · Jul 26, 2024 · d85a358 · d85a358
1 parent 57f6f48
commit d85a358
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/ai/backend/common/docker.py b/src/ai/backend/common/docker.py
@@ -234,8 +234,7 @@ async def login(
     """
     basic_auth: Optional[aiohttp.BasicAuth]
 
-    # TODO: Fix this.
-    if "public.ecr" in str(registry_url):
+    if "public.ecr" in str(registry_url) or "dkr.ecr" in str(registry_url):
         return {
             "auth": None,
             "headers": {