In order for the <figure> to take up the full width of the element that it is in, add the attribute data-size="full".

Content inside of the <figure> will stretch to take up all the available space.

Dog holding a stick.
This figure and it's content is taking up the full space it can.