We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5300eb6 commit dc5b8e7Copy full SHA for dc5b8e7
1 file changed
stubs/sklearn/utils/_set_output.pyi
@@ -4,4 +4,4 @@ from ..base import BaseEstimator
4
5
class _SetOutputMixin:
6
def __init_subclass__(cls, auto_wrap_output_keys: None | tuple[str] = ..., **kwargs) -> None: ...
7
- def set_output(self, *, transform: None | Literal["default", "pandas"] = None) -> BaseEstimator: ...
+ def set_output(self, *, transform: None | Literal["default", "pandas", "polars"] = None) -> BaseEstimator: ...
0 commit comments