Include PEP8 fixes only in graph classes
Revert "Include PEP8 fixes only in graph classes"

This reverts commit 200a26ccd86630d0e8d2fbf0db333bbf55d41b73.