File size: 565 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import datasets
from ..folder_based_builder import folder_based_builder
logger = datasets.utils.logging.get_logger(__name__)
class PdfFolderConfig(folder_based_builder.FolderBasedBuilderConfig):
"""BuilderConfig for ImageFolder."""
drop_labels: bool = None
drop_metadata: bool = None
def __post_init__(self):
super().__post_init__()
class PdfFolder(folder_based_builder.FolderBasedBuilder):
BASE_FEATURE = datasets.Pdf
BASE_COLUMN_NAME = "pdf"
BUILDER_CONFIG_CLASS = PdfFolderConfig
EXTENSIONS: list[str] = [".pdf"]
|