from cheesechaser.datapool import IncrementIDDataPool

_AIBOORU_REPO_ID = 'deepghs/aibooru-webp-4Mpixel'


class AIBooruWebpDataPool(IncrementIDDataPool):
    def __init__(self, revision: str = 'main'):
        IncrementIDDataPool.__init__(
            self,
            data_repo_id=_AIBOORU_REPO_ID,
            data_revision=revision,
            idx_repo_id=_AIBOORU_REPO_ID,
            idx_revision=revision,
            base_level=3,
        )