class_docstr: ARS Builder.
docstr: Create a dataset iterator to use for learning/updating the agent.
docstr_tok: ["Create","a","dataset","iterator","to","use","for","learning\/updating","the","agent","."]
code: def make_dataset_iterator(
      self, replay_client: reverb.Client
  ) -> Iterator[reverb.ReplaySample]:
      """Create a dataset iterator to use for learning/updating the agent."""
      dataset = reverb.TrajectoryDataset.from_table_signature(
          server_address=replay_client.server_address,
          table=self._config.replay_table_name,
          max_in_flight_samples_per_worker=1,
      )
      return utils.device_put(
          dataset.as_numpy_iterator(), jax.devices()[0]
      )
  
imports: from typing import Dict, Iterator, List, Optional, Tuple
  import acme
  from acme import adders
  from acme import core
  from acme import specs
  from acme.adders import reverb
  from acme.agents.jax import actor_core
  from acme.agents.jax import actors
  from acme.agents.jax import builders
  from acme.agents.jax.ars import config
  from acme.agents.jax.ars import learning
  from acme.jax import networks
  from acme.jax import running_statistics
  from acme.jax import utils
  from acme.jax import variable_utils
  from acme.utils import counting
  from acme.utils import loggers
  import jax
  import jax.numpy
  import numpy
  import reverb
label_desc: Use image node_id 5 for calling the ARSBuilder obj's underlying member method code with example usage: obj.make_dataset_iterator(replay_client) and returns: utils