I decided to go with an optional `ax=` argument instead that will defer to `plt.gca()` if it isn't specified.