BUG: train_val appears in data availability plot in legend

Currently the train_val subset appears in the legend of the data availability plot (but not in the cummulative one).

Check out what is going wrong and fix it.