allow `last_step` option in `generate_diagnostic_plots`