I am trying to run pystan 3.2 on our new python docker container. Python version == 3.8.8. I ran the code below and I keep getting a timeout error during the stan.build portion of the code. I used the FAQ to figure out that I should be using nest-asyncio and that is working fine, now its just timeout in the loop. The error ends here:
I am sure I messed up something in the set up so any help would be appreciated!
import stan
schools_code = """
data {
int<lower=0> J; // number of schools
real y[J]; // estimated treatment effects
real<lower=0> sigma[J]; // standard error of effect estimates
}
parameters {
real mu; // population treatment effect
real<lower=0> tau; // standard deviation in treatment effects
vector[J] eta; // unscaled deviation from mu by school
}
transformed parameters {
vector[J] theta = mu + tau * eta; // school treatment effects
}
model {
target += normal_lpdf(eta | 0, 1); // prior log-density
target += normal_lpdf(y | theta, sigma); // log-likelihood
}
"""
schools_data = {"J": 8,
"y": [28, 8, -3, 7, -1, 1, 18, 12],
"sigma": [15, 10, 16, 11, 9, 11, 10, 18]}
posterior = stan.build(schools_code, data=schools_data, random_seed=1)
## Not getting to these steps due to error above
fit = posterior.sample(num_chains=4, num_samples=100)
eta = fit["eta"] # array with shape (8, 4000)
df = fit.to_frame()