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)