Error when trying to plot the curves of the posteriors of lighweight_mmm model

Question

I'm working on a lightweight marketing mix modeling model but I get this error when I try to execute:

plot.plot_response_curves(media_mix_model=mmm, target_scaler=target_scaler)

Cannot interpret value of type <class 'lightweight_mmm.lightweight_mmm.LightweightMMM'> as an abstract array; it does not have a dtype attribute 

here is the link to the notebook : https://colab.research.google.com/drive/1JjFFXVsni3KQ73bSjiutZ9tLjQlcI652#scrollTo=0jRI8ynqzFip

I tried to get this :

mmm = lightweight_mmm.LightweightMMM(model_name="carryover")
number_warmup=100
number_samples=100

mmm.fit(
    media=media_data_train,
    media_prior=costs2,
    target=target_train,
    number_warmup=number_warmup,
    number_samples=number_samples,
    number_chains=1,
    ) 
 plot.plot_response_curves(media_mix_model=mmm, target_scaler=target_scaler)