Change remove_image to check if image is a dictionary and use Id value if it is.

1 files changed, 2 insertions, 0 deletions

diff --git a/docker/client.py b/docker/client.py
index 1ff43cc..14c9780 100644
--- a/docker/client.py
+++ b/docker/client.py
@@ -794,6 +794,8 @@ class Client(requests.Session):
         self._raise_for_status(res)
 
     def remove_image(self, image, force=False, noprune=False):
+        if isinstance(image, dict):
+            image = image.get('Id') 
         params = {'force': force, 'noprune': noprune}
         res = self._delete(self._url("/images/" + image), params=params)
         self._raise_for_status(res)