add changelog and fixed errors in test_Stratified_mandatory_y_argument

Anurag-Varma · Anurag-Varma · commit f24c4c9982dd · 2024-07-05T21:44:52.000+05:30
diff --git a/doc/whats_new/v1.6.rst b/doc/whats_new/v1.6.rst
@@ -182,6 +182,10 @@ Changelog
   estimator without re-fitting it.
   :pr:`29067` by :user:`Guillaume Lemaitre <glemaitre>`.
 
+- |Fix| :class:`model_selection.RepeatedStratifiedKFold` has the function split,
+  where 'y' parameter is made as mandatory input field.
+  :pr:`29402` by :user:`Anurag Varma <Anurag-Varma>`.
+
 :mod:`sklearn.neighbors`
 ........................
 
diff --git a/sklearn/model_selection/tests/test_split.py b/sklearn/model_selection/tests/test_split.py
@@ -2061,18 +2061,12 @@ def test_Stratified_mandatory_y_argument():
     # y = [0] * 90 + [1] * 10
     # Variable 'y' is required for Stratified classes split function.
     # Testing the split without 'y', shoudl raise TypeError.
+    msg = r"split\(\) missing 1 required positional argument: 'y'"
 
     cv = StratifiedKFold(n_splits=2)
-    msg = r".*StratifiedKFold\.split\(\) missing 1 required positional argument: 'y'.*"
     with pytest.raises(TypeError, match=msg):
-        for _i, (_train, _test) in enumerate(cv.split(x)):
-            pass
+        cv.split(x)
 
     cv = RepeatedStratifiedKFold(n_splits=2, n_repeats=3)
-    msg = (
-        r".*RepeatedStratifiedKFold\.split\(\) "
-        r"missing 1 required positional argument: 'y'.*"
-    )
     with pytest.raises(TypeError, match=msg):
-        for _i, (_train, _test) in enumerate(cv.split(x)):
-            pass
+        cv.split(x)
