From 60443ca258db222a232e2a24a77d0a3870fbdce8 Mon Sep 17 00:00:00 2001
From: Adam Dangoor <adamdangoor@gmail.com>
Date: Tue, 25 Jun 2024 14:23:03 +0100
Subject: [PATCH] Add more keyword argument types for Docker pull (#12210)

---
 stubs/docker/docker/api/image.pyi     |  5 +++--
 stubs/docker/docker/models/images.pyi | 30 ++++++++++++++++++++++++---
 2 files changed, 30 insertions(+), 5 deletions(-)

diff --git a/stubs/docker/docker/api/image.pyi b/stubs/docker/docker/api/image.pyi
index f274a2338..034bd633b 100644
--- a/stubs/docker/docker/api/image.pyi
+++ b/stubs/docker/docker/api/image.pyi
@@ -1,4 +1,5 @@
 from _typeshed import Incomplete
+from typing import Any
 
 log: Incomplete
 
@@ -39,9 +40,9 @@ class ImageApiMixin:
         repository: str,
         tag: str | None = None,
         stream: bool = False,
-        auth_config: Incomplete | None = None,
+        auth_config: dict[str, Any] | None = None,
         decode: bool = False,
-        platform: Incomplete | None = None,
+        platform: str | None = None,
         all_tags: bool = False,
     ): ...
     def push(
diff --git a/stubs/docker/docker/models/images.pyi b/stubs/docker/docker/models/images.pyi
index b88a8b6be..279bd6dcf 100644
--- a/stubs/docker/docker/models/images.pyi
+++ b/stubs/docker/docker/models/images.pyi
@@ -75,11 +75,35 @@ class ImageCollection(Collection[Image]):
     def list(self, name: str | None = None, all: bool = False, filters: dict[str, Any] | None = None) -> _ImageList: ...
     def load(self, data: bytes) -> _ImageList: ...
     @overload
-    def pull(self, repository: str, tag: str | None = None, all_tags: Literal[False] = False, **kwargs) -> Image: ...
+    def pull(
+        self,
+        repository: str,
+        tag: str | None = None,
+        all_tags: Literal[False] = False,
+        *,
+        platform: str | None = None,
+        auth_config: dict[str, Any] | None = None,
+    ) -> Image: ...
     @overload
-    def pull(self, repository: str, tag: str | None = None, *, all_tags: Literal[True], **kwargs) -> _ImageList: ...
+    def pull(
+        self,
+        repository: str,
+        tag: str | None = None,
+        *,
+        all_tags: Literal[True],
+        auth_config: dict[str, Any] | None = None,
+        platform: str | None = None,
+    ) -> _ImageList: ...
     @overload
-    def pull(self, repository: str, tag: str | None, all_tags: Literal[True], **kwargs) -> _ImageList: ...
+    def pull(
+        self,
+        repository: str,
+        tag: str | None,
+        all_tags: Literal[True],
+        *,
+        auth_config: dict[str, Any] | None = None,
+        platform: str | None = None,
+    ) -> _ImageList: ...
     def push(self, repository: str, tag: str | None = None, **kwargs): ...
     def remove(self, *args, **kwargs) -> None: ...
     def search(self, *args, **kwargs): ...