MNT refactoring in routing _MetadataRequester (#31534)

adrinjalali · OmarManzoor · StefanieSenger · web-flow · commit 1c5c8f03659d · 2025-09-05T09:55:25.000Z
Co-authored-by: Omar Salman &lt;omar.salman2007@gmail.com&gt;
Co-authored-by: Stefanie Senger &lt;91849487+StefanieSenger@users.noreply.github.com&gt;
diff --git a/examples/miscellaneous/plot_metadata_routing.py b/examples/miscellaneous/plot_metadata_routing.py
@@ -167,7 +167,7 @@ def get_metadata_routing(self):
         # This method defines the routing for this meta-estimator.
         # In order to do so, a `MetadataRouter` instance is created, and the
         # routing is added to it. More explanations follow below.
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self.estimator,
             method_mapping=MethodMapping()
             .add(caller="fit", callee="fit")
@@ -352,7 +352,7 @@ def __init__(self, estimator):
 
     def get_metadata_routing(self):
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             # defining metadata routing request values for usage in the meta-estimator
             .add_self_request(self)
             # defining metadata routing request values for usage in the sub-estimator
@@ -483,7 +483,7 @@ def __init__(self, transformer, classifier):
 
     def get_metadata_routing(self):
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             # We add the routing for the transformer.
             .add(
                 transformer=self.transformer,
@@ -613,7 +613,7 @@ def fit(self, X, y, **fit_params):
         self.estimator_ = clone(self.estimator).fit(X, y, **routed_params.estimator.fit)
 
     def get_metadata_routing(self):
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self.estimator,
             method_mapping=MethodMapping().add(caller="fit", callee="fit"),
         )
@@ -650,7 +650,7 @@ def fit(self, X, y, sample_weight=None, **fit_params):
 
     def get_metadata_routing(self):
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             .add_self_request(self)
             .add(
                 estimator=self.estimator,
diff --git a/sklearn/calibration.py b/sklearn/calibration.py
@@ -578,7 +578,7 @@ def get_metadata_routing(self):
             routing information.
         """
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             .add_self_request(self)
             .add(
                 estimator=self._get_estimator(),
diff --git a/sklearn/compose/_column_transformer.py b/sklearn/compose/_column_transformer.py
@@ -1289,7 +1289,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__)
+        router = MetadataRouter(owner=self)
         # Here we don't care about which columns are used for which
         # transformers, and whether or not a transformer is used at all, which
         # might happen if no columns are selected for that transformer. We
diff --git a/sklearn/compose/_target.py b/sklearn/compose/_target.py
@@ -382,7 +382,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             regressor=self._get_regressor(),
             method_mapping=MethodMapping()
             .add(caller="fit", callee="fit")
diff --git a/sklearn/covariance/_graph_lasso.py b/sklearn/covariance/_graph_lasso.py
@@ -1138,7 +1138,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             splitter=check_cv(self.cv),
             method_mapping=MethodMapping().add(callee="split", caller="fit"),
         )
diff --git a/sklearn/ensemble/_bagging.py b/sklearn/ensemble/_bagging.py
@@ -636,7 +636,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__)
+        router = MetadataRouter(owner=self)
 
         method_mapping = MethodMapping()
         method_mapping.add(caller="fit", callee="fit").add(
diff --git a/sklearn/ensemble/_stacking.py b/sklearn/ensemble/_stacking.py
@@ -397,7 +397,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__)
+        router = MetadataRouter(owner=self)
 
         # `self.estimators` is a list of (name, est) tuples
         for name, estimator in self.estimators:
diff --git a/sklearn/ensemble/_voting.py b/sklearn/ensemble/_voting.py
@@ -180,7 +180,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__)
+        router = MetadataRouter(owner=self)
 
         # `self.estimators` is a list of (name, est) tuples
         for name, estimator in self.estimators:
diff --git a/sklearn/feature_selection/_from_model.py b/sklearn/feature_selection/_from_model.py
@@ -498,7 +498,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self.estimator,
             method_mapping=MethodMapping()
             .add(caller="partial_fit", callee="partial_fit")
diff --git a/sklearn/feature_selection/_rfe.py b/sklearn/feature_selection/_rfe.py
@@ -551,7 +551,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self.estimator,
             method_mapping=MethodMapping()
             .add(caller="fit", callee="fit")
@@ -1002,7 +1002,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__)
+        router = MetadataRouter(owner=self)
         router.add(
             estimator=self.estimator,
             method_mapping=MethodMapping().add(caller="fit", callee="fit"),
diff --git a/sklearn/feature_selection/_sequential.py b/sklearn/feature_selection/_sequential.py
@@ -353,7 +353,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__)
+        router = MetadataRouter(owner=self)
         router.add(
             estimator=self.estimator,
             method_mapping=MethodMapping().add(caller="fit", callee="fit"),
diff --git a/sklearn/impute/_iterative.py b/sklearn/impute/_iterative.py
@@ -1023,7 +1023,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self.estimator,
             method_mapping=MethodMapping().add(callee="fit", caller="fit"),
         )
diff --git a/sklearn/linear_model/_coordinate_descent.py b/sklearn/linear_model/_coordinate_descent.py
@@ -1949,7 +1949,7 @@ def get_metadata_routing(self):
             routing information.
         """
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             .add_self_request(self)
             .add(
                 splitter=check_cv(self.cv),
diff --git a/sklearn/linear_model/_least_angle.py b/sklearn/linear_model/_least_angle.py
@@ -1821,7 +1821,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             splitter=check_cv(self.cv),
             method_mapping=MethodMapping().add(caller="fit", callee="split"),
         )
diff --git a/sklearn/linear_model/_logistic.py b/sklearn/linear_model/_logistic.py
@@ -2305,7 +2305,7 @@ def get_metadata_routing(self):
         """
 
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             .add_self_request(self)
             .add(
                 splitter=self.cv,
diff --git a/sklearn/linear_model/_omp.py b/sklearn/linear_model/_omp.py
@@ -1121,7 +1121,7 @@ def get_metadata_routing(self):
             routing information.
         """
 
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             splitter=self.cv,
             method_mapping=MethodMapping().add(caller="fit", callee="split"),
         )
diff --git a/sklearn/linear_model/_ransac.py b/sklearn/linear_model/_ransac.py
@@ -707,7 +707,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self.estimator,
             method_mapping=MethodMapping()
             .add(caller="fit", callee="fit")
diff --git a/sklearn/linear_model/_ridge.py b/sklearn/linear_model/_ridge.py
@@ -2502,7 +2502,7 @@ def get_metadata_routing(self):
             routing information.
         """
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             .add_self_request(self)
             .add(
                 scorer=self._get_scorer(),
diff --git a/sklearn/metrics/_scorer.py b/sklearn/metrics/_scorer.py
@@ -218,7 +218,7 @@ def get_metadata_routing(self):
             A :class:`~utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        return MetadataRouter(owner=self.__class__.__name__).add(
+        return MetadataRouter(owner=self).add(
             **self._scorers,
             method_mapping=MethodMapping().add(caller="score", callee="score"),
         )
@@ -274,6 +274,9 @@ def __repr__(self):
             f"{response_method_string}{kwargs_string})"
         )
 
+    def _routing_repr(self):
+        return repr(self)
+
     def __call__(self, estimator, X, y_true, sample_weight=None, **kwargs):
         """Evaluate predicted target values for X relative to y_true.
 
@@ -363,7 +366,7 @@ def set_score_request(self, **kwargs):
             ),
             kwargs=kwargs,
         )
-        self._metadata_request = MetadataRequest(owner=self.__class__.__name__)
+        self._metadata_request = MetadataRequest(owner=self)
         for param, alias in kwargs.items():
             self._metadata_request.score.add_request(param=param, alias=alias)
         return self
@@ -494,7 +497,10 @@ def __call__(self, estimator, *args, **kwargs):
         return estimator.score(*args, **kwargs)
 
     def __repr__(self):
-        return f"{self._estimator.__class__}.score"
+        return f"{type(self._estimator).__name__}.score"
+
+    def _routing_repr(self):
+        return repr(self)
 
     def _accept_sample_weight(self):
         # TODO(slep006): remove when metadata routing is the only way
diff --git a/sklearn/model_selection/_classification_threshold.py b/sklearn/model_selection/_classification_threshold.py
@@ -392,7 +392,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self.estimator,
             method_mapping=MethodMapping().add(callee="fit", caller="fit"),
         )
@@ -858,7 +858,7 @@ def get_metadata_routing(self):
             routing information.
         """
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             .add(
                 estimator=self.estimator,
                 method_mapping=MethodMapping().add(callee="fit", caller="fit"),
diff --git a/sklearn/model_selection/_search.py b/sklearn/model_selection/_search.py
@@ -1215,7 +1215,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__)
+        router = MetadataRouter(owner=self)
         router.add(
             estimator=self.estimator,
             method_mapping=MethodMapping().add(caller="fit", callee="fit"),
diff --git a/sklearn/multiclass.py b/sklearn/multiclass.py
@@ -624,7 +624,7 @@ def get_metadata_routing(self):
         """
 
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             .add_self_request(self)
             .add(
                 estimator=self.estimator,
@@ -1028,7 +1028,7 @@ def get_metadata_routing(self):
         """
 
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             .add_self_request(self)
             .add(
                 estimator=self.estimator,
@@ -1277,7 +1277,7 @@ def get_metadata_routing(self):
             routing information.
         """
 
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self.estimator,
             method_mapping=MethodMapping().add(caller="fit", callee="fit"),
         )
diff --git a/sklearn/multioutput.py b/sklearn/multioutput.py
@@ -330,7 +330,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self.estimator,
             method_mapping=MethodMapping()
             .add(caller="partial_fit", callee="partial_fit")
@@ -1149,7 +1149,7 @@ def get_metadata_routing(self):
             routing information.
         """
 
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self._get_estimator(),
             method_mapping=MethodMapping().add(caller="fit", callee="fit"),
         )
@@ -1311,7 +1311,7 @@ def get_metadata_routing(self):
             routing information.
         """
 
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self._get_estimator(),
             method_mapping=MethodMapping().add(caller="fit", callee="fit"),
         )
diff --git a/sklearn/pipeline.py b/sklearn/pipeline.py
@@ -1340,7 +1340,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__)
+        router = MetadataRouter(owner=self)
 
         # first we add all steps except the last one
         for _, name, trans in self._iter(with_final=False, filter_passthrough=True):
@@ -2103,7 +2103,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__)
+        router = MetadataRouter(owner=self)
 
         for name, transformer in self.transformer_list:
             router.add(
diff --git a/sklearn/semi_supervised/_self_training.py b/sklearn/semi_supervised/_self_training.py
@@ -601,7 +601,7 @@ def get_metadata_routing(self):
             A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
             routing information.
         """
-        router = MetadataRouter(owner=self.__class__.__name__)
+        router = MetadataRouter(owner=self)
         router.add(
             estimator=self.estimator,
             method_mapping=(
diff --git a/sklearn/tests/metadata_routing_common.py b/sklearn/tests/metadata_routing_common.py
@@ -491,7 +491,7 @@ def fit(self, X, y, **fit_params):
         self.estimator_ = clone(self.estimator).fit(X, y, **params.estimator.fit)
 
     def get_metadata_routing(self):
-        router = MetadataRouter(owner=self.__class__.__name__).add(
+        router = MetadataRouter(owner=self).add(
             estimator=self.estimator,
             method_mapping=MethodMapping().add(caller="fit", callee="fit"),
         )
@@ -520,7 +520,7 @@ def predict(self, X, **predict_params):
 
     def get_metadata_routing(self):
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             .add_self_request(self)
             .add(
                 estimator=self.estimator,
@@ -550,7 +550,7 @@ def fit(self, X, y, sample_weight=None, **kwargs):
 
     def get_metadata_routing(self):
         router = (
-            MetadataRouter(owner=self.__class__.__name__)
+            MetadataRouter(owner=self)
             .add_self_request(self)
             .add(
                 estimator=self.estimator,
@@ -576,7 +576,7 @@ def transform(self, X, y=None, **transform_params):
         return self.transformer_.transform(X, **params.transformer.transform)
 
     def get_metadata_routing(self):
-        return MetadataRouter(owner=self.__class__.__name__).add(
+        return MetadataRouter(owner=self).add(
             transformer=self.transformer,
             method_mapping=MethodMapping()
             .add(caller="fit", callee="fit")
diff --git a/sklearn/tests/test_metadata_routing.py b/sklearn/tests/test_metadata_routing.py
diff --git a/sklearn/utils/_metadata_requests.py b/sklearn/utils/_metadata_requests.py
diff --git a/sklearn/utils/tests/test_estimator_checks.py b/sklearn/utils/tests/test_estimator_checks.py

Original file line number	Diff line number	Diff line change
`@@ -1138,7 +1138,7 @@ def get_metadata_routing(self):`
`1138`	`1138`	A :class:`~sklearn.utils.metadata_routing.MetadataRouter` encapsulating
`1139`	`1139`	`routing information.`
`1140`	`1140`	`"""`
`1141`		`- router = MetadataRouter(owner=self.__class__.__name__).add(`
	`1141`	`+ router = MetadataRouter(owner=self).add(`
`1142`	`1142`	`splitter=check_cv(self.cv),`
`1143`	`1143`	`method_mapping=MethodMapping().add(callee="split", caller="fit"),`
`1144`	`1144`	`)`