code: def testGetEpisodeListsSomeRows(self):
      spec = self.default_specs()
      replay_table = episodic_table.EpisodicTable(spec, capacity=3)
      empty_values = self.np_values(spec, 0)
      test_values = self.np_values(spec, 5)
      self.evaluate(tf.compat.v1.global_variables_initializer())
      self.evaluate(replay_table.append(1, test_values))
      # Get episode lists for rows 1 and 2. The returned values should look like:
      # {'action': [<tensorlist for row 1>, <tensorlist for row 2>],
      #  'camera': [<tensorlist for row 1>, <tensorlist for row 2>],
      #  'lidar': [<tensorlist for row 1>, <tensorlist for row 2>]}
      episode_lists = replay_table.get_episode_lists([1, 2])
      for episode_list_slot in tf.nest.flatten(episode_lists):
          self.assertEqual(episode_list_slot.shape.rank, 1)
          self.assertEqual(
              self.evaluate(tf.size(input=episode_list_slot)), 2
          )
      # Stack episode tensors for row 1, i.e.:
      # {'action': <five items>, 'camera': <five items>, 'lidar': <five items>}
      episode_tensors_1 = tf.nest.map_structure(
          lambda slot, lists: replay_table._stack_tensor_list(
              slot, lists[0]
          ),
          replay_table.slots,
          episode_lists,
      )
      # Should be equivalent to
      # Stack episode tensors for row 2, which is empty.
      episode_tensors_2 = tf.nest.map_structure(
          lambda slot, lists: replay_table._stack_tensor_list(
              slot, lists[1]
          ),
          replay_table.slots,
          episode_lists,
      )
      self.assertAllClose(self.evaluate(episode_tensors_1), test_values)
      self.assertAllClose(
          self.evaluate(episode_tensors_2), empty_values
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import numpy
  import tensorflow
  from tf_agents import specs
  from tf_agents.replay_buffers import episodic_table
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 11 for calling the EpisodicTableTest obj's underlying member method code with example usage: obj.testGetEpisodeListsSomeRows() without return types