What is the best practice for resizing vega lite graph with column property?

Question

The code below allows me to resize, but it fails if I use any graph that have a column property. Also, I feel like my current method is very hacky and may not be the correct