Revisiting Bayesian Survival Analysis in Python with PyMC

revisit_bayes_survival.ipynb

Hi.
Thank you. I gratefully appreciate the effort.
Having identical env after triggering the 23th cell I have TypingError: Failed in nopython mode pipeline (step: nopython frontend)
Bugtrace connects it to numba. I have numba==0.56.4. Please attach the numba version you've used.

Thanks for the kind words @wojla

> conda list | grep numba
numba                     0.56.4                   pypi_0    pypi

Thank you.
The issue is within nutpie.
I used a standard NUTS sampling form pymc
Instead:

trace = nutpie.sample(
    nutpie.compile_pymc_model(model),
    seed=SEED
)

I called:

with model:
    trace = pm.sample()

It solved all my issues.