DOC Ensures that KBinsDiscretizer passes numpydoc validation (scikit-learn#21016)

genvalen · adrinjalali · commit 73dec986e999 · 2021-09-20T10:16:00.000+02:00
diff --git a/maint_tools/test_docstrings.py b/maint_tools/test_docstrings.py
@@ -16,7 +16,6 @@
     "HuberRegressor",
     "Isomap",
     "IterativeImputer",
-    "KBinsDiscretizer",
     "KNNImputer",
     "LabelPropagation",
     "LabelSpreading",
diff --git a/sklearn/preprocessing/_discretization.py b/sklearn/preprocessing/_discretization.py
@@ -133,7 +133,6 @@ class KBinsDiscretizer(TransformerMixin, BaseEstimator):
            [-0.5,  2.5, -2.5, -0.5],
            [ 0.5,  3.5, -1.5,  0.5],
            [ 0.5,  3.5, -1.5,  1.5]])
-
     """
 
     def __init__(self, n_bins=5, *, encode="onehot", strategy="quantile", dtype=None):
@@ -157,7 +156,8 @@ def fit(self, X, y=None):
 
         Returns
         -------
-        self
+        self : object
+            Returns the instance itself.
         """
         X = self._validate_data(X, dtype="numeric")
 
