Seems to need a failing cell to execute and save an image. Ideally would have the interactive version working, not sure if this can be done (altair doesn't natively support treemaps)