Skip to content

Commit dc5b8e7

Browse files
authored
sklearn: Allow 'polars' in set_output (#394)
1 parent 5300eb6 commit dc5b8e7

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

stubs/sklearn/utils/_set_output.pyi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,4 @@ from ..base import BaseEstimator
44

55
class _SetOutputMixin:
66
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: ...
7+
def set_output(self, *, transform: None | Literal["default", "pandas", "polars"] = None) -> BaseEstimator: ...

0 commit comments

Comments
 (0)