def splits(self, root, train_iterable=False):
        """Get the train and test dataset from the folder

        Args:
            root (str): Directory where the data can be loaded.

        Returns:
            Tuple of ExtSumProcessedIterableDataset as train dataset
            and ExtSumProcessedDataset as test dataset.
        """
        train_files, test_files = self._get_files(root)
        if train_iterable:
            return (
                ExtSumProcessedIterableDataset(train_files, is_shuffle=True),
                ExtSumProcessedDataset(test_files, is_shuffle=False),
            )
        else:
            return (
                ExtSumProcessedDataset(train_files, is_shuffle=True),
                ExtSumProcessedDataset(test_files, is_shuffle=False),
            )
Get the train and test dataset from the folder

        Args:
            root (str): Directory where the data can be loaded.

        Returns:
            Tuple of ExtSumProcessedIterableDataset as train dataset
            and ExtSumProcessedDataset as test dataset.
        
utils_nlp/models/transformers/extractive_summarization.py