You should inherit from sklearn.base.BaseEstimator