Plotly Express LaTeX in VS Code

I can't typeset a Plotly Express label with LaTeX style using VS Code. It just shows the string as is. Example:

import plotly.express as px
fig = px.density_heatmap(df, x="x", y="y", z="z", title="$x^2$")

Do you have any hints?