code: def test_await_type_python(self):
      def foo() -> Tensor:
          return torch.randn()
  
      awaits = torch.jit.annotate(List[Await[Tensor]], [])
      awaits.append(torch.jit._awaitable(foo))
  
imports: import io
  import torch
  from torch.testing._internal.jit_utils import JitTestCase
  from torch.testing._internal.jit_utils import make_global
  from typing import List, Optional, Tuple
  from torch import Tensor
  from torch._awaits import _Await
label_desc: Use image node_id 2 for calling the TestAwait obj's underlying member method code with example usage: obj.test_await_type_python() and returns: torch