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

arisayosh · samronsin · commit ab6552825fed · 2021-11-30T17:28:07.000+01: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
@@ -2326,7 +2326,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
@@ -2359,7 +2359,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.
@@ -2410,7 +2409,6 @@ def train_test_split(
 
     >>> train_test_split(y, shuffle=False)
     [[0, 1, 2], [3, 4]]
-
     """
     n_arrays = len(arrays)
     if n_arrays == 0:
