The endpoint for obtaining data for the Sankey diagram needs to be fixed because the query to find all transitions doesn't take into account the other phases between training phases. See the picture, line 93: