i will also leave up the branch I pushed to demonstrate the workaround around the matplotlib bug that this exposed