Skip to content

Commit 66c94a2

Browse files
authored
[sklearn] Make BaseEstimator inherit from _MetadataRequester
1 parent d1b4fb3 commit 66c94a2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

stubs/sklearn/base.pyi

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ from ._config import get_config as get_config
1515
from ._typing import ArrayLike, Float, Int, MatrixLike
1616
from .metrics import accuracy_score as accuracy_score, r2_score as r2_score
1717
from .utils._estimator_html_repr import estimator_html_repr as estimator_html_repr
18+
from .utils._metadata_requests import _MetadataRequester
1819
from .utils._param_validation import validate_parameter_constraints as validate_parameter_constraints
1920
from .utils._set_output import _SetOutputMixin
2021
from .utils.validation import check_array as check_array, check_is_fitted as check_is_fitted, check_X_y as check_X_y
@@ -24,7 +25,7 @@ from .utils.validation import check_array as check_array, check_is_fitted as che
2425

2526
def clone(estimator: BaseEstimator | Iterable[BaseEstimator], *, safe: bool = True) -> Any: ...
2627

27-
class BaseEstimator:
28+
class BaseEstimator(_MetadataRequester):
2829
def get_params(self, deep: bool = True) -> dict: ...
2930
def set_params(self, **params) -> Self: ...
3031
def __repr__(self, N_CHAR_MAX: int = 700) -> str: ...

0 commit comments

Comments
 (0)