DOC Ensures that assert_all_finite passes numpydoc validation (#22470)

UgoM · thomasjpfan · glemaitre · web-flow · commit 9712e80bbc0c · 2022-02-14T10:45:07.000+01:00
Co-authored-by: Thomas J. Fan &lt;thomasjpfan@gmail.com&gt;
Co-authored-by: Guillaume Lemaitre &lt;g.lemaitre58@gmail.com&gt;
diff --git a/sklearn/tests/test_docstrings.py b/sklearn/tests/test_docstrings.py
@@ -172,7 +172,6 @@
     "sklearn.utils.sparsefuncs.mean_variance_axis",
     "sklearn.utils.sparsefuncs.min_max_axis",
     "sklearn.utils.tosequence",
-    "sklearn.utils.validation.assert_all_finite",
     "sklearn.utils.validation.check_is_fitted",
     "sklearn.utils.validation.check_memory",
     "sklearn.utils.validation.check_random_state",
diff --git a/sklearn/utils/validation.py b/sklearn/utils/validation.py
@@ -156,8 +156,10 @@ def assert_all_finite(
     Parameters
     ----------
     X : {ndarray, sparse matrix}
+        The input data.
 
     allow_nan : bool, default=False
+        If True, do not throw error when `X` contains NaN.
 
     estimator_name : str, default=None
         The estimator name, used to construct the error message.
