remove PlotComponent abstract base class

Author: drbenvincent

causalpy/plotting.py

@@ -11,7 +11,6 @@
 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
-from abc import ABC
 
 import arviz as az
 import matplotlib.pyplot as plt
@@ -25,13 +24,7 @@
 az.style.use("arviz-darkgrid")
 
 
-class PlotComponent(ABC):
-    """Abstract Base Class for PlotComponent."""
-
-    pass
-
-
-class BayesianPlotComponent(PlotComponent):
+class BayesianPlotComponent:
     """Plotting component for Bayesian models."""
 
     @staticmethod
@@ -443,7 +436,7 @@ def plot_regression_kink(results, round_to=None):
         return fig, ax
 
 
-class OLSPlotComponent(PlotComponent):
+class OLSPlotComponent:
     """Plotting component for OLS models."""
 
     @staticmethod

causalpy/pymc_models.py

@@ -23,7 +23,7 @@
 import xarray as xr
 from arviz import r2_score
 
-from causalpy.plotting import BayesianPlotComponent, PlotComponent
+from causalpy.plotting import BayesianPlotComponent
 from causalpy.utils import round_num
 
 
@@ -159,7 +159,7 @@ def calculate_impact(self, y_true, y_pred):
     def calculate_cumulative_impact(self, impact):
         return impact.cumsum(dim="obs_ind")
 
-    def get_plot_component(self) -> PlotComponent:
+    def get_plot_component(self) -> BayesianPlotComponent:
         return BayesianPlotComponent()
 
     def print_coefficients(self, labels, round_to=None) -> None:

causalpy/skl_models.py

@@ -21,7 +21,7 @@
 from sklearn.linear_model import LinearRegression
 from sklearn.linear_model._base import LinearModel
 
-from causalpy.plotting import OLSPlotComponent, PlotComponent
+from causalpy.plotting import OLSPlotComponent
 from causalpy.utils import round_num
 
 
@@ -36,7 +36,7 @@ def calculate_cumulative_impact(self, impact):
         """Calculate the cumulative impact intervention."""
         return np.cumsum(impact)
 
-    def get_plot_component(self) -> PlotComponent:
+    def get_plot_component(self) -> OLSPlotComponent:
         """Get the plot component type for the model."""
         return OLSPlotComponent()