[jit] Fix pretty printing of functions (#37432)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37432
Fixes https://github.com/pytorch/pytorch/issues/36803.
Test Plan: Imported from OSS
Differential Revision: D21284735
Pulled By: suo
fbshipit-source-id: 8c673099b3171070bff80fd1defc91487f66d4b3
