docstr: generator that iterates over the dataset
  
  :param batch_size: (int) the size of the batch
  :return: (dict) a batch of the input data of size 'batch_size'
docstr_tok: ["generator","that","iterates","over","the","dataset",":","param","batch_size",":","(","int",")","the","size","of","the","batch",":","return",":","(","dict",")","a","batch","of","the","input","data","of","size","'batch_size","'"]
code: def iterate_once(self, batch_size):
      """
      generator that iterates over the dataset
  
      :param batch_size: (int) the size of the batch
      :return: (dict) a batch of the input data of size 'batch_size'
      """
      if self.shuffle:
          self.shuffle_dataset()
  
      while self._next_id <= self.n_samples - batch_size:
          yield self.next_batch(batch_size)
      self._next_id = 0
  
file_path: stable-baselines/stable_baselines/common/dataset.py
imports: import numpy
label_desc: Use image node_id 4 for calling the Dataset obj's underlying member method code with example usage: obj.iterate_once(batch_size) without return types