def _fetch_block(self, start_ds_idx, offset, end_ds_idx, length):
        """
        Fetch a block of tokens based on its dataset idx
        """
        buffer = torch.cat(
            [self.dataset[idx] for idx in range(start_ds_idx, end_ds_idx + 1)]
        )
        s, e = offset, offset + length
        return buffer[s:e]

        Fetch a block of tokens based on its dataset idx
        
structured_prompting/fairseq-version/fairseq/fairseq/data/legacy/block_pair_dataset.py