import matplotlib.pyplot as plt
from pymc_marketing.mmm.events import AsymmetricGaussianBasis
from pymc_extras.prior import Prior
asy_gaussian = AsymmetricGaussianBasis(
    priors={
        "sigma_before": Prior("Gamma", mu=[3, 4], sigma=1, dims="event"),
        "a_after": Prior("Normal", mu=[-.75, .5], sigma=.2, dims="event"),
    }
)
coords = {"event": ["PyData-Berlin", "PyCon-Finland"]}
prior = asy_gaussian.sample_prior(coords=coords)
curve = asy_gaussian.sample_curve(prior)
fig, axes = asy_gaussian.plot_curve(
    curve, subplot_kwargs={"figsize": (6, 3), "sharey": True}
)
for ax in axes:
    ax.set_xlabel("")
plt.show()