Heatmapgl traces in Python