# 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 NeelnandaCounterfactTracingRecord(DatasetSchema):
    DATASET_NAME: ClassVar[str] = "NeelNanda/counterfact-tracing"
    CONFIG_NAME: ClassVar[Optional[str]] = None
    relation: str
    relation_prefix: str
    relation_suffix: str
    prompt: str
    relation_id: str
    target_false_id: str
    target_true_id: str
    target_true: str
    target_false: str
    subject: str
