from solar_eval.models.base import CustomBaseModel


class DatasetInfo(CustomBaseModel):
    repo: str
    path: str
    size: int | None = None
    num_samples: int | None = None
    cached: bool = False


class DatasetSyncRequest(CustomBaseModel):
    repo: str
    force: bool = False