Different renderer for summary text and axis labels

[benjaminpope]

Hi Sam,

I notice there is a different renderer for the summary text and the axis labels.

Defining labels like this in numpyro

    dra, ddec, log_flux = [npy.sample(r"\Delta \text{RA}",    dist.Uniform(100, 200)), 
              npy.sample(r"\Delta \text{Dec}",   dist.Uniform(100, 200)),
              npy.sample(r"\log \text{Flux}",    dist.Uniform(-5,-3))
            ]

I get math mode not rendered in the axis labels

but if I just use unicode on the labels

    dra, ddec, log_flux = [npy.sample("Δ RA",    dist.Uniform(100, 200)), 
              npy.sample("Δ Dec",   dist.Uniform(100, 200)),
              npy.sample("log Flux",    dist.Uniform(-5,-3))
            ]

I get

where the summary text is now displayed in math mode, which is also not what we want.

Is it possible to make these render consistently?

Cheers,

Ben

[Samreay]

Hmmmm yeah that's strange, I'll look into this when I'm back from parental leave. Please ping me here if I haven't comments again in a week or two. Similarly, if you look into it yourself and figure it out, a full or partial PR will gladly be accepted:)