From e7901f994a07432853b3c0aff4196b813bdec9fd Mon Sep 17 00:00:00 2001 From: Maren Westermann Date: Tue, 10 May 2022 12:55:24 +0200 Subject: [PATCH 1/2] correct return type of check_random_state --- sklearn/utils/validation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sklearn/utils/validation.py b/sklearn/utils/validation.py index e5a96ce622ba9..c47c1f6e92b0a 100644 --- a/sklearn/utils/validation.py +++ b/sklearn/utils/validation.py @@ -1171,8 +1171,8 @@ def check_random_state(seed): Returns ------- - None - No returns. + ``RandomState`` + The appropriate ``RandomState`` based on the input. """ if seed is None or seed is np.random: return np.random.mtrand._rand From b1ee917dff899dfe5f5678c1f1cff08dc9712862 Mon Sep 17 00:00:00 2001 From: Maren Westermann Date: Thu, 12 May 2022 16:03:44 +0200 Subject: [PATCH 2/2] implement suggested change --- sklearn/utils/validation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sklearn/utils/validation.py b/sklearn/utils/validation.py index c47c1f6e92b0a..879ddac3e6f42 100644 --- a/sklearn/utils/validation.py +++ b/sklearn/utils/validation.py @@ -1171,8 +1171,8 @@ def check_random_state(seed): Returns ------- - ``RandomState`` - The appropriate ``RandomState`` based on the input. + :class:`numpy:numpy.random.RandomState` + The random state object based on `seed` parameter. """ if seed is None or seed is np.random: return np.random.mtrand._rand