Skip to content

Commit dfb31ee

Browse files
farhanreynaldotwiecki
authored andcommitted
fix trace_to_dataframe test
1 parent e07b3b6 commit dfb31ee

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pymc/tests/test_sampling.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -767,9 +767,10 @@ def test_deterministic_of_observed_modified_interface(self):
767767
return_inferencedata=False,
768768
compute_convergence_checks=False,
769769
)
770-
ppc_trace = pm.trace_to_dataframe(
771-
trace, varnames=[n for n in trace.varnames if n != "out"]
772-
).to_dict("records")
770+
varnames = [v for v in trace.varnames if v != "out"]
771+
ppc_trace = [
772+
dict(zip(varnames, row)) for row in zip(*(trace.get_values(v) for v in varnames))
773+
]
773774
ppc = pm.sample_posterior_predictive(
774775
return_inferencedata=False,
775776
model=model,

0 commit comments

Comments
 (0)