Commit 6e69df75 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

To use _gettags you need to extend BaseEstimator

parent 6736926a
......@@ -7,7 +7,7 @@ import cloudpickle
import functools
from sklearn.preprocessing import FunctionTransformer
from sklearn.base import TransformerMixin
from sklearn.base import TransformerMixin, BaseEstimator
from sklearn.pipeline import Pipeline
from dask import delayed
import dask.bag
......@@ -191,7 +191,7 @@ def _make_kwargs_from_samples(samples, arg_attr_list):
return kwargs
class SampleMixin:
class SampleMixin(BaseEstimator):
"""Mixin class to make scikit-learn estimators work in :any:`Sample`-based
Do not use this class except for scikit-learn estimators.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment