Skip to content

Commit c6717c4

Browse files
committed
Inherit from BaseEstimator
1 parent cad794b commit c6717c4

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

‎crowdkit/aggregation/base/__init__.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,12 +10,13 @@
1010

1111
import attr
1212
import pandas as pd
13+
from sklearn.base import BaseEstimator
1314

1415
from ..utils import named_series_attrib
1516

1617

1718
@attr.s
18-
class BaseClassificationAggregator:
19+
class BaseClassificationAggregator(BaseEstimator): # type: ignore[misc]
1920
"""This is a base class for all classification aggregators
2021
2122
Attributes:

0 commit comments

Comments
 (0)