From 7dc7a2659af3081c0e2c1edcc74a8daacf924477 Mon Sep 17 00:00:00 2001
From: DHUK98 <32871749+DHUK98@users.noreply.github.com>
Date: Wed, 3 Jun 2020 18:11:21 +0100
Subject: [PATCH] Corrected output of image __str__ to include height

---
 visual_genome/models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/visual_genome/models.py b/visual_genome/models.py
index 4321ad4..b921304 100644
--- a/visual_genome/models.py
+++ b/visual_genome/models.py
@@ -21,12 +21,12 @@ def __init__(self, id, url, width, height, coco_id, flickr_id):
         self.flickr_id = flickr_id
 
     def __str__(self):
-        return 'id: %d, coco_id: %d, flickr_id: %d, width: %d, url: %s' \
+        return 'id: %d, coco_id: %d, flickr_id: %d, width: %d, height: %d, url: %s' \
             % (self.id, -1
                 if self.coco_id is None
                 else self.coco_id, -1
                 if self.flickr_id is None
-                else self.flickr_id, self.width, self.url)
+                else self.flickr_id, self.width, self.height, self.url)
 
     def __repr__(self):
         return str(self)