PyMC’s pm.sample() now returns an arviz InferenceData object by default. So your trace variable is already what you want. Of course, this assumes you are running a recent version of PyMC (e.g., > 5.0).
PyMC’s pm.sample() now returns an arviz InferenceData object by default. So your trace variable is already what you want. Of course, this assumes you are running a recent version of PyMC (e.g., > 5.0).