DOC Ensures that MiniBatchKMeans passes numpydoc validation (#21202)

Pinky-Chaudhary · web-flow · commit 8bb1a43e6a8f · 2021-09-30T10:02:36.000+02:00
diff --git a/maint_tools/test_docstrings.py b/maint_tools/test_docstrings.py
@@ -14,7 +14,6 @@
     "LabelSpreading",
     "LocalOutlierFactor",
     "LocallyLinearEmbedding",
-    "MiniBatchKMeans",
     "MultiLabelBinarizer",
     "MultiTaskElasticNet",
     "MultiTaskElasticNetCV",
diff --git a/sklearn/cluster/_kmeans.py b/sklearn/cluster/_kmeans.py
@@ -1133,7 +1133,7 @@ def fit(self, X, y=None, sample_weight=None):
 
         Returns
         -------
-        self
+        self : object
             Fitted estimator.
         """
         X = self._validate_data(
@@ -1885,7 +1885,8 @@ def fit(self, X, y=None, sample_weight=None):
 
         Returns
         -------
-        self
+        self : object
+            Fitted estimator.
         """
         X = self._validate_data(
             X,
@@ -2038,7 +2039,8 @@ def partial_fit(self, X, y=None, sample_weight=None):
 
         Returns
         -------
-        self
+        self : object
+            Return updated estimator.
         """
         has_centers = hasattr(self, "cluster_centers_")
 
