only one type of plot -> inside ArrheniusRegressor

MANIFEST.in

@@ -2,10 +2,10 @@ include LICENSE
 include README.md
 include pyproject.toml
 include sierras.py
+include test_sierras.py
 
 exclude requirements_dev.txt
 exclude tox.ini
 exclude .readthedocs.yml
 
-recursive-exclude tests *
 recursive-exclude docs *

sierras.py

@@ -68,8 +68,7 @@
 # CONSTANTS
 # =============================================================================
 
-__all__ = ["ArrheniusRegressor", "ArrheniusPlotter"]
-
+__all__ = ["ArrheniusRegressor"]
 
 NAME = "sierras"
 
@@ -188,26 +187,8 @@ def to_dataframe(self, X, y, sample_weight=None):
 
         return df
 
-    @property
-    def plot(self):
-        """Arrhenius plot accessor."""
-        return ArrheniusPlotter(self)
-
-
-class ArrheniusPlotter:
-    """Arrhenius plot.
-
-    Parameters
-    ----------
-    areg : sierras.ArrheniusRegressor
-        An ArrheniusRegressor already fitted.
-    """
-
-    def __init__(self, areg):
-        self.areg = areg
-
-    def arrhenius(self, X, y, ax=None, data_kws=None, pred_kws=None):
-        """Arrhenius plot function.
+    def plot(self, ax=None, data_kws=None, pred_kws=None):
+        """Arrhenius plotter.
 
         Parameters
         ----------
@@ -237,12 +218,7 @@ def arrhenius(self, X, y, ax=None, data_kws=None, pred_kws=None):
 
         pred_kws.setdefault("label", "fit")
 
-        ax.errorbar(
-            self.areg._X,
-            self.areg._y,
-            yerr=self.areg._sample_weight,
-            **data_kws,
-        )
-        ax.plot(self.areg._X, self.areg.reg_.predict(self.areg._X), **pred_kws)
+        ax.errorbar(self._X, self._y, yerr=self._sample_weight, **data_kws)
+        ax.plot(self._X, self.reg_.predict(self._X), **pred_kws)
 
         return ax

test_sierras.py

@@ -129,7 +129,7 @@ def test_plot(self, fig_test, fig_ref, dataset, request):
 
         # test
         test_ax = fig_test.subplots()
-        areg.plot.arrhenius(X, y, ax=test_ax)
+        areg.plot(ax=test_ax)
 
         # expected
         ref_ax = fig_ref.subplots()