From @wyli: "this is hard-coded to yaml and a style, perhaps we make it a separate method and support more options cc @mingxin-zheng"

Originally posted by @mingxin-zheng in #5240 (comment)