Thanks! But I guess what I really want is both {prior} and {trace} packaged into a single object. In current state, they both have different components:

But I know that it’s possible to have a single object with everything consolidated inside. For example, here is a single InferenceData object that has everything together.

Maybe this question has nothing to do with az.from_pymc.