[Lint] Auto format graph_module.py (#108594)
Summary: Auto format the `graph_module.py` file
Test Plan: lint
Differential Revision: D48983066
Pull Request resolved: https://github.com/pytorch/pytorch/pull/108594
Approved by: https://github.com/jiayisuse