for s in self.dataset.take(self.load_max_docs)
            if self.sample_to_document_function is not None
        )
[docs]    def load(self) -> List[Document]:
        """Download a selected dataset.
        Returns: a list of Documents.
        """
        return list(self.lazy_load())