MAINT Update _kmeans.py warning message (#26362)

hmaarrfk · web-flow · commit a2b85716728d · 2023-05-16T09:31:10.000-04:00
diff --git a/sklearn/cluster/_kmeans.py b/sklearn/cluster/_kmeans.py
@@ -894,6 +894,7 @@ def _check_params_vs_input(self, X, default_n_init=None):
                     " explicitly to suppress the warning"
                 ),
                 FutureWarning,
+                stacklevel=2,
             )
             self._n_init = default_n_init
         if self._n_init == "auto":

Original file line number	Diff line number	Diff line change
`@@ -894,6 +894,7 @@ def _check_params_vs_input(self, X, default_n_init=None):`
`894`	`894`	`" explicitly to suppress the warning"`
`895`	`895`	`),`
`896`	`896`	`FutureWarning,`
	`897`	`+ stacklevel=2,`
`897`	`898`	`)`
`898`	`899`	`self._n_init = default_n_init`
`899`	`900`	`if self._n_init == "auto":`