Support covariates in `plot_expected_purchases`

[ColtAllen]

Issue created to resume work started by @PabloRoque in the now-closed #1430.

[PabloRoque]

@ColtAllen I had to erase my fork because it was consuming Github Actions credits due to cron jobs