rename plot accessors

probabl-ai · Jan 3, 2025 · 2df34d1 · 2df34d1
1 parent 4b23afb
commit 2df34d1
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/skore/src/skore/sklearn/_estimator.py b/skore/src/skore/sklearn/_estimator.py
@@ -477,11 +477,11 @@ def _get_X_y_and_data_source_hash(self, *, data_source, X=None, y=None):
 
 
 ########################################################################################
-# Plotting accessor
+# Plotting accessor for metrics
 ########################################################################################
 
 
-class _PlotAccessor(_BaseAccessor):
+class _PlotMetricsAccessor(_BaseAccessor):
     def __init__(self, parent):
         # Note: parent here will be the MetricsAccessor instance
         super().__init__(parent._parent, icon="🎨")
@@ -756,6 +756,9 @@ def prediction_error(
 
         return display
 
+    def _get_help_title(self):
+        return f"{self._icon} Available plot methods"
+
 
 ###############################################################################
 # Metrics accessor
@@ -780,7 +783,7 @@ class _MetricsAccessor(_BaseAccessor):
     def __init__(self, parent):
         super().__init__(parent, icon="📏")
         # Create plot sub-accessor
-        self.plot = _PlotAccessor(self)
+        self.plot = _PlotMetricsAccessor(self)
 
     # TODO: should build on the `add_scorers` function
     def report_metrics(