Notebooks don't work as-is

For example, the lightgbm example notebook has:

from hummingbird.ml import convert

and then

hb_model = convert(model, 'pytorch')

This returns an error:

In [14]: hb_model = convert(model, 'pytorch')
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-14-5658f007aeed> in <module>
----> 1 hb_model = convert(model, 'pytorch')

TypeError: 'module' object is not callable

Should it instead be: from hummingbird.ml.convert import convert_lightgbm as convert?

[jspisak]

Hey Folks - i still see the NBs failing after pip installing HummingBird. Is there an ETA to update PyPI?

[ksaur]

pushed https://pypi.org/project/hummingbird-ml/0.0.2/