code: def sample_transitions(self, batch_size, batch_info) -> Transitions:
          
          probs, indexes, tree_idxs = self.backend.sample_batch_transitions(batch_size)
  
          return self._get_transitions(probs, indexes, tree_idxs, batch_info)
docs: Sample batch of transitions and return them