code: def test_enum_return(self):
      class Color(Enum):
          RED = 1
          GREEN = 2
  
      make_global(Color)
  
      @torch.jit.script
      def return_enum(cond: bool):
          if cond:
              return Color.RED
          else:
              return Color.GREEN
  
      self.assertEqual(return_enum(True), Color.RED)
      self.assertEqual(return_enum(False), Color.GREEN)
  
imports: import os
  import sys
  import torch
  from torch.testing import FileCheck
  from enum import Enum
  from typing import Any, List
  from torch.testing._internal.jit_utils import JitTestCase, make_global
label_desc: Use image node_id 13 for calling the TestEnum obj's underlying member method code with example usage: obj.test_enum_return() and returns: Color, Color