code: def get_random_observations_rewards_actions_dones(
      self, batch_size=None
  ):
      batch_size = batch_size or self.BATCH_SIZE
      # Random observations, rewards, actions, done of the expected shape.
      observations = np.random.rand(
          *((batch_size,) + self.OBSERVATION_SHAPE)
      )
      raw_rewards = np.random.randn(batch_size)
      actions = np.random.randn(batch_size)
      # 40% change of being done.
      dones = np.random.random((batch_size,)) > 0.6
  
      return observations, raw_rewards, actions, dones
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import numpy
  from tensor2tensor.envs import time_step
  from tensor2tensor.envs import trajectory
  import tensorflow.compat.v1
  from tensorflow.compat.v1.io import gfile
label_desc: Use image node_id 1 for calling the BatchTrajectoryTest obj's underlying member method code with example usage: obj.get_random_observations_rewards_actions_dones(batch_size) and returns: observations,  raw_rewards,  actions,  dones