src/generate/schema.py

from typing import Any

import pandera.pandas as pa


class GeneratedDatasetSchema(pa.DataFrameModel):
    id_: pa.typing.Series[int] = pa.Field(alias="id")
    generated_answer: pa.typing.Series[Any]