import matplotlib.pyplot as plt
import numpy as np
from pymc_marketing.mmm import RootSaturation

rng = np.random.default_rng(0)

saturation = RootSaturation()
prior = saturation.sample_prior(random_seed=rng)
curve = saturation.sample_curve(prior)
saturation.plot_curve(curve, random_seed=rng)
plt.show()