code: def test_build_model_beam(self):
      inputs, targets, outputs, _ = visualization.build_model(
          hparams_set,
          model_name,
          self.data_dir,
          problem_name,
          beam_size=8,
      )
  
      self.assertAllEqual((1, None, 1, 1), inputs.shape.as_list())
      self.assertAllEqual((1, None, 1, 1), targets.shape.as_list())
      self.assertAllEqual((None, None), outputs.shape.as_list())
  
file_path: tensor2tensor/tensor2tensor/visualization/visualization_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  from tensor2tensor.utils import trainer_lib
  from tensor2tensor.visualization import visualization
  import tensorflow.compat.v1
label_desc: Use image node_id 3 for calling the VisualizationTest obj's underlying member method code with example usage: obj.test_build_model_beam() without return types