fixed heatmap args in graph pipeline)