code: def __init__(self, output_vocab_size: int):
      self.output_vocab_size = output_vocab_size
      self.states = []
  
imports: from typing import List, Optional
  from typing import Tuple
  import numpy
  import pytest
  import torch
  import numpy
  import sockeye.beam_search
  import sockeye.constants
  import sockeye.lexicon
  import sockeye.utils
label_desc: Use image node_id 1 to create a new _TestInference object from inherited base classes: sockeye.beam_search with example: obj = _TestInference(output_vocab_size)