Add `yaml` to installation dependency in setup.py (#5044)
* Add `yaml` to installation dependency in setup.py
Fixes https://github.com/pytorch/xla/issues/4871
---
Add `yaml` to installation dependency in setup.py. Local build succeeds. I'll let CI verify the rest.
* Update to pyyaml