src/display/utils.py

from dataclasses import dataclass


@dataclass(frozen=True)
class ColumnSpec:
    name: str
    type: str
    displayed_by_default: bool
    hidden: bool = False
    never_hidden: bool = False


MODEL_INFO_COLUMNS = [
    ColumnSpec("T", "str", True, never_hidden=True),
    ColumnSpec("Model", "markdown", True, never_hidden=True),
    ColumnSpec("Type", "str", False, hidden=True),
    ColumnSpec("Precision", "str", False, hidden=True),
    ColumnSpec("Organization", "str", True),
    ColumnSpec("Test Leak.", "str", True),
    ColumnSpec("Replication Code", "str", True),
]


def fields(_cls):
    return MODEL_INFO_COLUMNS