Add pyyaml requirement

(cherry picked from commit 3e1060da68c947b28b0e2e64cdd2421e840e62d3)