This post seems to be a good solution, but, as it was said in that post, we would need to overwrite the data in the trace object to swap the sampled values for the switched dimensions. Is it possible to edit the data of an Arviz data structure?
I want to do something like:
idata2 = idata.copy()
idata2.sel(chain=[1]).posterior['mu'][0][:,0] = idata2.sel(chain=[1]).posterior['mu'][0][:,1]
But, by doing this, idata2 is unaltered.