pytorch

9063cb0a - Infer types for arguments of methods not invoked directly by monkeytype (#57202)

Infer types for arguments of methods not invoked directly by monkeytype (#57202) Summary: Support adding type annotations for class methods and nn.Module methods which are not invoked under the hood of MonkeyType ** Changes ** * This PR involves a slight change in how the example inputs are passed while scripting `class` and `nn.Module` objects. * The example inputs passed to `_script_pdt` is of the following format: - example_inputs= [(obj.method1, (arg_list)), (obj.method2, (arg_list)),] * For nn.Modules, to infer types for `forward` methods, example_inputs can be passed in two ways: - example_inputs= [(obj.forward, (arg_list, ))] - example_inputs = [(obj, (arg_list, ) )] Pull Request resolved: https://github.com/pytorch/pytorch/pull/57202 Reviewed By: desertfire Differential Revision: D28382827 Pulled By: nikithamalgifb fbshipit-source-id: 5481467f3e909493bf3f439ee312056943508534