Sample batch of transitions and return them

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

        return self._get_transitions(probs, indexes, tree_idxs, batch_info)