Setting Plotly Heatmap Range

I am trying to plot an animation of heatmaps with plotly and I want them to have the same legend range. Is there an easy way to do that, similar to setting vmin/vmax range in Seaborn?

# this works in seaborn
ax = sns.heatmap(data[0], vmin = vmin, vmax = vmax) 

# my plotly code
figure = {
       'data': [trace],
       'layout': {},
       'frames': frames
     }