scikit-learn
diff --git a/‎sklearn/feature_selection/_mutual_info.py
Copy file name to clipboardExpand all lines: sklearn/feature_selection/_mutual_info.py
+10Lines changed: 10 additions & 0 deletions b/‎sklearn/feature_selection/_mutual_info.py
Copy file name to clipboardExpand all lines: sklearn/feature_selection/_mutual_info.py
+10Lines changed: 10 additions & 0 deletions
diff --git a/‎sklearn/tests/test_public_functions.py
Copy file name to clipboardExpand all lines: sklearn/tests/test_public_functions.py
+1Lines changed: 1 addition & 0 deletions b/‎sklearn/tests/test_public_functions.py
Copy file name to clipboardExpand all lines: sklearn/tests/test_public_functions.py
+1Lines changed: 1 addition & 0 deletions
@@ -311,6 +311,16 @@ def _estimate_mi(
     return np.array(mi)
 
 
+@validate_params(
+    {
+        "X": ["array-like", "sparse matrix"],
+        "y": ["array-like"],
+        "discrete_features": [StrOptions({"auto"}), "boolean", "array-like"],
+        "n_neighbors": [Interval(Integral, 1, None, closed="left")],
+        "copy": ["boolean"],
+        "random_state": ["random_state"],
+    }
+)
 def mutual_info_regression(
     X, y, *, discrete_features="auto", n_neighbors=3, copy=True, random_state=None
 ):
 
@@ -125,6 +125,7 @@ def _check_function_param_validation(
     "sklearn.feature_selection.f_classif",
     "sklearn.feature_selection.f_regression",
     "sklearn.feature_selection.mutual_info_classif",
+    "sklearn.feature_selection.mutual_info_regression",
     "sklearn.feature_selection.r_regression",
     "sklearn.linear_model.orthogonal_mp",
     "sklearn.metrics.accuracy_score",