fix pep8

Signed-off-by: Prasanna Santhanam <tsp@apache.org>