Better types in fbcode/caffe2/torch/jit/_script.py (#55856)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55856
Test Plan: Sandcastle
Reviewed By: SplitInfinity
Differential Revision: D27715495
fbshipit-source-id: 9804e2d432fda302117f05a0d21cbb7f0dd3ae38
