Formatting: Corrected formatting for the files using google yapf styling

[sanidhyamangal]

There were several .py files not formatted correctly so updated the formatting using yapf.