PyMC v5.0.1 - tutorials in docs page not working from the conda environment (Windows 11)

Questions v5
5

Hi,

Thanks for the suggestion,
Just wrapped it on the if __name__ == '__main__': statement, like so:

import pandas as pd
import numpy as np

print('*** Start script ***')

if __name__ == '__main__':

    disaster_data = pd.Series(
    [4, 5, 4, 0, 1, 4, 3, 4, 0, 6, 3, 3, 4, 0, 2, 6,
    3, 3, 5, 4, 5, 3, 1, 4, 4, 1, 5, 5, 3, 4, 2, 5,
    2, 2, 3, 4, 2, 1, 3, np.nan, 2, 1, 1, 1, 1, 3, 0, 0,
    1, 0, 1, 1, 0, 0, 3, 1, 0, 3, 2, 2, 0, 1, 1, 1,
    0, 1, 0, 1, 0, 0, 0, 2, 1, 0, 0, 0, 1, 1, 0, 2,
    3, 3, 1, np.nan, 2, 1, 1, 1, 1, 2, 4, 2, 0, 0, 1, 4,
    0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 1])

    years = np.arange(1851, 1962)

    with pm.Model() as disaster_model:

        switchpoint = pm.DiscreteUniform("switchpoint", lower=years.min(), upper=years.max())

        early_rate = pm.Exponential("early_rate", 1.0)
        late_rate = pm.Exponential("late_rate", 1.0)

        # Allocate appropriate Poisson rates to years before and after current
        rate = pm.math.switch(switchpoint >= years, early_rate, late_rate)

        disasters = pm.Poisson("disasters", rate, observed=disaster_data)

    with disaster_model:
        idata = pm.sample(10000)

and I get same error as above.
for chains =1, it is able to sample,
for chains>1, get the same error as above.

for any number of cores, and chains=1 it samples.
for any number of cores, and chains>1 it does not sample.

Is there anything I could do on Windows to reset any settings that might be causing this, or something along those lines? Thank you