[BUILD] Fix PEP 8