[BUG]: `plot_delay_time_distribution` not working

[maxbriel]

Describe the bug
plot_delay_time_distribution does not use the model_weights yet and needs to be updated to be usable again.

[sgossage]

Since we merged PR#735, is this still an issue?