# Auto-generated by schemas-hf. Do not edit manually.

from dataclasses import dataclass
from typing import ClassVar, Optional

from schemas_hf.base import DatasetSchema
from schemas_hf.registry import register_schema


@register_schema
@dataclass
class UwunionArcagiRecord(DatasetSchema):
    DATASET_NAME: ClassVar[str] = "uwunion/arcagi"
    CONFIG_NAME: ClassVar[Optional[str]] = None
    rid: int
    task_id: str
    split: str
    pair_type: str
    pair_index: int
    input_json: str
    output_json: str
    input_height: int
    input_width: int
    output_height: int
    output_width: int
    input_cells: int
    output_cells: int
    input_n_colors: int
    output_n_colors: int
    input_color_hist: str
    output_color_hist: str
