Az.plot_trace() gets an "No module named 'pymc3'" Error

Questions v5
,
1

print(pm.__version__)
5.0.1
print(az.__version__)
0.14.0

trace = pm.sample(...)
az.plot_trace(trace)

gives me the following Error:

ModuleNotFoundError: No module named 'pymc3'

It looks like arziv is still looking for pymc3 to convert a pymc trace object to its xarray.Dataset object, according to arviz.convert_to_dataset — ArviZ 0.14.0 documentation

Any response will be warmly appreciated.

Regards,

2

Welcome!

Are you calling pm.sample() with return_inferencedata=False? Can you check out the type of trace? It should be of type InferenceData.

3

Thanks for your prompt reply!

Yes, my return_inferencedata was set to False. Switching it to True solved my problem.

Fantastic, thanks a lot!

1 Like