Migrate to support Arviz 1.0 #620
Description
Arviz is preparing for a backwards-compatibility breaking 1.0 release. We should migrate pymc-extras to support it.
The migration guide seems pretty decent as a place to start for that. Though, I suspect we do this piecemeal between the example notebooks and the codebase itself.