I am using 3.11.4.
reset_coords gave an error (ValueError: cannot remove index coordinates with reset_coords: {'draw'}) but I just tried reset_index and that worked just fine! Thank you very much!
I think I am going to have to invest some time into learning Arviz / xdata data structures !