class_docstr: Base helper class for next frame tests.
code: def TestVideoModelWithActionAndRewardsInfer(
      self, in_frames, out_frames, hparams, model, expected_last_dim
  ):
      del expected_last_dim
      hparams = fill_hparams(hparams, in_frames, out_frames)
      hparams = full_modalities(hparams)
      hparams.reward_prediction = True
  
      features = create_full_features(in_frames, out_frames)
  
      output = self.InferModel(model, hparams, features)
  
      self.assertTrue(isinstance(output, dict))
      self.assertTrue("outputs" in output.keys())
      self.assertTrue("scores" in output.keys())
      self.assertTrue("targets" in output.keys())
      self.assertTrue("target_reward" in output.keys())
      expected_shape = get_tensor_shape(features["targets"])
      self.assertEqual(output["targets"].shape, expected_shape)
      expected_shape = get_tensor_shape(features["target_reward"])[:2]
      self.assertEqual(output["target_reward"].shape, expected_shape)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  from tensor2tensor.data_generators import video_generated
  from tensor2tensor.layers import modalities
  from tensor2tensor.utils import registry
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
label_desc: Use image node_id 8 for calling the BaseNextFrameTest obj's underlying member method code with example usage: obj.TestVideoModelWithActionAndRewardsInfer(in_frames, out_frames, hparams, model, expected_last_dim) without return types