Pm.sample result does not work with az.plot_trace or other az functions

Yes! I was about to ask this. Your explanation actually explains everything now.