export plot data from plot utils

lpoug · lpoug · commit bb7305adfd3d · 2025-02-07T11:57:43.000+01:00
diff --git a/causalpy/plot_utils.py b/causalpy/plot_utils.py
@@ -24,6 +24,7 @@
 import xarray as xr
 from matplotlib.collections import PolyCollection
 from matplotlib.lines import Line2D
+from sklearn.base import RegressorMixin
 
 
 def plot_xY(
