code: def get_arg_return_types_from_interface(module_interface):
      assert getattr(
          module_interface, "__torch_script_interface__", False
      ), "Expect a TorchScript class interface decorated by @torch.jit.interface."
      qualified_name = torch._jit_internal._qualified_name(
          module_interface
      )
      cu = torch.jit._state._python_cu
      module_interface_c = cu.get_interface(qualified_name)
      assert (
          "forward" in module_interface_c.getMethodNames()
      ), f"Expect forward in interface methods, while it has {module_interface_c.getMethodNames()}"
      method_schema = module_interface_c.getMethod("forward")
  
      arg_str_list = []
      arg_type_str_list = []
      assert method_schema is not None
      for argument in method_schema.arguments:
          arg_str_list.append(argument.name)
  
          if argument.has_default_value():
              default_value_str = f" = {argument.default_value}"
          else:
              default_value_str = ""
          arg_type_str = (
              f"{argument.name}: {argument.type}{default_value_str}"
          )
          arg_type_str_list.append(arg_type_str)
  
      arg_str_list = arg_str_list[1:]  # Remove "self".
      args_str = ", ".join(arg_str_list)
  
      arg_type_str_list = arg_type_str_list[1:]  # Remove "self".
      arg_types_str = ", ".join(arg_type_str_list)
  
      assert len(method_schema.returns) == 1
      argument = method_schema.returns[0]
      return_type_str = str(argument.type)
  
      return args_str, arg_types_str, return_type_str
  
file_path: pytorch/torch/distributed/nn/jit/instantiator.py
imports: import importlib
  import logging
  import os
  import sys
  import tempfile
  from typing import Optional
  import torch
  from torch.distributed.nn.jit.templates.remote_module_template import get_remote_module_template
label_desc: Use image node_id 1 for calling a global function with example usage: get_arg_return_types_from_interface(module_interface) and returns: args_str,  arg_types_str,  return_type_str