This MR removes the matplotlib pin that was added in !1293 (merged) to address a CI failure, if the CI still passes, I suggest getting this in.
