Fix type mismatches in io_pyjags.py. (#1498)
Fix type mismatches in io_pyjags.py. (#1498)

* Fix io_pyjags.py types.

* Fix log_likelihood type hints and remove use of requires.

* Fix incorrect type hint for save_warmup function parameter.
arviz-devs/arviz,arviz-devs/arviz,arviz-devs/arviz,arviz-devs/arviz