Vega + Sankey Diagram Color Change

You could try and use this example as a guide: