Skip to content

Commit 27cd067

Browse files
fonnesbecktwiecki
authored andcommitted
Fixed KeyError in trace_to_dataframe (#2122)
1 parent bfe63c2 commit 27cd067

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc3/backends/tracetab.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def trace_to_dataframe(trace, chains=None, varnames=None, include_transformed=Fa
3434
flat_names = {v: create_flat_names(v, var_shapes[v]) for v in varnames}
3535

3636
var_dfs = []
37-
for v in var_shapes:
37+
for v in varnames:
3838
vals = trace.get_values(v, combine=True, chains=chains)
3939
flat_vals = vals.reshape(vals.shape[0], -1)
4040
var_dfs.append(pd.DataFrame(flat_vals, columns=flat_names[v]))

0 commit comments

Comments
 (0)