Skip to content

Commit 39ddaf0

Browse files
🐛 Update varname param to var_names for arviz dep
1 parent bae7f98 commit 39ddaf0

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

examples/case_studies/disaster_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,4 +152,4 @@
152152
start = {"early_mean": 2.0, "late_mean": 3.0}
153153

154154
tr = pm.sample(1000, tune=500, start=start)
155-
pm.plot_trace(tr)
155+
az.plot_trace(tr)

examples/pymc3_howto/LKJ_correlation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def run(n=1000):
5454
with model:
5555
trace = pm.sample(n)
5656
az.plot_trace(
57-
trace, varnames=["mu", "r"], lines={"mu": mu_r, "r": corr_r[np.triu_indices(n_var, k=1)]}
57+
trace, var_names=["mu", "r"], lines={"mu": mu_r, "r": corr_r[np.triu_indices(n_var, k=1)]}
5858
)
5959

6060

examples/pymc3_howto/rankdata_ordered.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,14 +46,14 @@ def run(n=1500):
4646
with m:
4747
trace = pm.sample(n)
4848

49-
az.plot_trace(trace, varnames=["mu_hat"])
49+
az.plot_trace(trace, var_names=["mu_hat"])
5050

5151
print("Example observed data: ")
5252
print(y[:30, :].T)
5353
print("The true ranking is: ")
5454
print(yreal.flatten())
5555
print("The Latent mean is: ")
56-
latentmu = np.hstack(([0], az.summary(trace, varnames=["mu_hat"])["mean"].values))
56+
latentmu = np.hstack(([0], az.summary(trace, var_names=["mu_hat"])["mean"].values))
5757
print(np.round(latentmu, 2))
5858
print("The estimated ranking is: ")
5959
print(np.argsort(latentmu))

0 commit comments

Comments
 (0)