fix: Silent failure of DockerAgent.push_image

lablup · Jul 29, 2024 · d2e8827 · d2e8827
1 parent 81dcb9a
commit d2e8827
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/ai/backend/agent/docker/agent.py b/src/ai/backend/agent/docker/agent.py
@@ -1343,7 +1343,13 @@ async def push_image(self, image_ref: ImageRef, registry_conf: ImageRegistry) ->
             }
 
         async with closing_async(Docker()) as docker:
-            await docker.images.push(image_ref.canonical, auth=auth_config)
+            result = await docker.images.push(image_ref.canonical, auth=auth_config)
+
+            # Why is this list? It contradicts the API documentation.
+            result_ = cast(list, result)
+
+            if error := result_[-1].get("error"):
+                raise RuntimeError(f"Failed to push image: {error}")
 
     async def pull_image(self, image_ref: ImageRef, registry_conf: ImageRegistry) -> None:
         auth_config = None