From 1c6d770e9b429dd8dd10ad65ea263f521ec7d062 Mon Sep 17 00:00:00 2001
From: FBruzzesi <francesco.bruzzesi.93@gmail.com>
Date: Mon, 20 Apr 2026 21:58:43 +0200
Subject: [PATCH] sklearn: Allow 'polars' in set_output

---
 stubs/sklearn/utils/_set_output.pyi | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/stubs/sklearn/utils/_set_output.pyi b/stubs/sklearn/utils/_set_output.pyi
index 234e7fc3..e9f6b919 100644
--- a/stubs/sklearn/utils/_set_output.pyi
+++ b/stubs/sklearn/utils/_set_output.pyi
@@ -4,4 +4,4 @@ from ..base import BaseEstimator
 
 class _SetOutputMixin:
     def __init_subclass__(cls, auto_wrap_output_keys: None | tuple[str] = ..., **kwargs) -> None: ...
-    def set_output(self, *, transform: None | Literal["default", "pandas"] = None) -> BaseEstimator: ...
+    def set_output(self, *, transform: None | Literal["default", "pandas", "polars"] = None) -> BaseEstimator: ...