feat: Apply pagination to ECR fetch_repositories

lablup · Jul 26, 2024 · d496c74 · d496c74
1 parent b15f9fc
commit d496c74
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ai/backend/manager/container_registry/aws_ecr.py b/src/ai/backend/manager/container_registry/aws_ecr.py
@@ -79,9 +79,9 @@ async def fetch_repositories(
         try:
             while True:
                 if next_token:
-                    response = client.describe_repositories(nextToken=next_token)
+                    response = client.describe_repositories(nextToken=next_token, maxResults=30)
                 else:
-                    response = client.describe_repositories()
+                    response = client.describe_repositories(maxResults=30)
 
                 for repo in response["repositories"]:
                     # TODO: Fix this.