draw_ellipse missing from astroML v1.0

[gtrichards]

I'm trying to run the example of Figure 6.6 (2nd edition: http://www.astroml.org/book_figures/chapter6/fig_EM_metallicity.html).

However, draw_ellipse seems to be missing. The error doesn't make sense to me:

ImportError: cannot import name 'LinearRegressionwithErrors' from 'astroML.linear_model' (/Users/gtr/opt/anaconda3/lib/python3.8/site-packages/astroML/linear_model/init.py)

However, I see "draw_ellipse" in tools.py at
~/anaconda3/lib/python3.8/site-packages/astroML/plotting

I managed to get it to work with by importing:
from matplotlib.patches import Ellipse

And then copying draw_elllipse from tools.py:
def draw_ellipse(mu, C, scales=[1, 2, 3], ax=None, **kwargs):