Update setup.py with PyYAML