DOC Ensures that train_test_split passes numpydoc validation (scikit-learn#21435)

arisayosh · ogrisel · commit 5a1810a1dafa · 2021-10-28T17:20:59.000+02:00
diff --git a/maint_tools/test_docstrings.py b/maint_tools/test_docstrings.py
@@ -172,7 +172,6 @@
     "sklearn.metrics.pairwise.rbf_kernel",
     "sklearn.metrics.pairwise.sigmoid_kernel",
     "sklearn.model_selection._split.check_cv",
-    "sklearn.model_selection._split.train_test_split",
     "sklearn.model_selection._validation.cross_val_score",
     "sklearn.model_selection._validation.cross_validate",
     "sklearn.model_selection._validation.learning_curve",
diff --git a/sklearn/model_selection/_split.py b/sklearn/model_selection/_split.py
@@ -2330,7 +2330,7 @@ def train_test_split(
     shuffle=True,
     stratify=None,
 ):
-    """Split arrays or matrices into random train and test subsets
+    """Split arrays or matrices into random train and test subsets.
 
     Quick utility that wraps input validation and
     ``next(ShuffleSplit().split(X, y))`` and application to input data
@@ -2363,7 +2363,6 @@ def train_test_split(
         Pass an int for reproducible output across multiple function calls.
         See :term:`Glossary <random_state>`.
 
-
     shuffle : bool, default=True
         Whether or not to shuffle the data before splitting. If shuffle=False
         then stratify must be None.
@@ -2414,7 +2413,6 @@ def train_test_split(
 
     >>> train_test_split(y, shuffle=False)
     [[0, 1, 2], [3, 4]]
-
     """
     n_arrays = len(arrays)
     if n_arrays == 0:
