ImportError: cannot import name 'get_dataset'

[sharthZ23]

Code:

from lightning.classification import AdaGradClassifier

Traceback:

ImportError                               Traceback (most recent call last)
<ipython-input-2-d9e7c6a3740a> in <module>()
     17 from sklearn.model_selection import GridSearchCV
     18 
---> 19 from lightning.classification import AdaGradClassifier
     20 
     21 import matplotlib.pyplot as plt

/usr/local/lib/python3.5/dist-packages/lightning/classification.py in <module>()
----> 1 from .impl.adagrad import AdaGradClassifier
      2 from .impl.primal_cd import CDClassifier
      3 from .impl.fista import FistaClassifier
      4 from .impl.sag import SAGClassifier
      5 from .impl.sag import SAGAClassifier

/usr/local/lib/python3.5/dist-packages/lightning/impl/adagrad.py in <module>()
      9 
     10 from .base import BaseClassifier, BaseRegressor
---> 11 from .dataset_fast import get_dataset
     12 from .adagrad_fast import _adagrad_fit
     13 

ImportError: cannot import name 'get_dataset'

Got on 0.5.0 and 0.6.dev0 versions.
watermark output:

CPython 3.5.2
IPython 6.2.1

numpy 1.13.3
scipy 1.0.0
sklearn 0.19.1

compiler   : GCC 5.4.0 20160609
system     : Linux
release    : 4.4.0-1054-aws
machine    : x86_64
processor  : x86_64
CPU cores  : 36
interpreter: 64bit