code: def _create_replay_buffer(self, rb_cls):
      self._stack_count = 4
      self._single_shape = (15, 15, 1)
      shape = (15, 15, self._stack_count)
      observation_spec = array_spec.ArraySpec(shape, np.int32, "obs")
      time_step_spec = ts.time_step_spec(observation_spec)
      action_spec = policy_step.PolicyStep(
          array_spec.BoundedArraySpec(
              shape=(),
              dtype=np.int32,
              minimum=0,
              maximum=1,
              name="action",
          )
      )
      self._trajectory_spec = trajectory.from_transition(
          time_step_spec, action_spec, time_step_spec
      )
  
      self._capacity = 32
      self._replay_buffer = rb_cls(
          data_spec=self._trajectory_spec, capacity=self._capacity
      )
  
imports: from __future__ import division
  from __future__ import unicode_literals
  import os
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tf_agents.replay_buffers import py_hashed_replay_buffer
  from tf_agents.replay_buffers import py_uniform_replay_buffer
  from tf_agents.specs import array_spec
  from tf_agents.trajectories import policy_step
  from tf_agents.trajectories import time_step
  from tf_agents.trajectories import trajectory
  from tf_agents.utils import nest_utils
label_desc: Use image node_id 1 for calling the PyUniformReplayBufferTest obj's underlying member method code with example usage: obj._create_replay_buffer(rb_cls) without return types