src/schemas/display_toml.py

from typing import Annotated

from pydantic import BaseModel, ConfigDict, Field


class DisplayToml(BaseModel):
    model_config = ConfigDict(extra="allow", frozen=True)

    version: Annotated[str, Field(..., description="The version of the results.")]
    benchmarks_order: Annotated[
        list[str],
        Field(
            default_factory=lambda: [
                "vsi_bench",
                "mmsi_bench",
                "mindcube_tiny",
                "viewspatial",
                "site",
            ],
            description="The predefined order of the benchmarks.",
        ),
    ]