env/config.py

"""TemporalBenchEnv configuration."""

from __future__ import annotations

from dataclasses import dataclass, field


DEFAULT_DOMAINS: tuple[str, ...] = ("PSML", "freshretailnet", "MIMIC", "causal_chambers")


@dataclass
class EnvConfig:
    """Episode structure, rewards, and data paths."""

    num_questions: int = 9
    primary_count: int = 6
    secondary_count: int = 3
    primary_domain: str = "PSML"
    allowed_task_types: tuple[str, ...] = ("T1U", "T3", "T2_MCQ")
    alpha: float = 1.0
    lambda_ep: float = 0.5
    partial_credit_t1u: bool = False
    case_insensitive_match: bool = False
    curriculum_stage: int = 3
    question_bank_path: str | None = None
    seed: int | None = None
    forecasting_enabled: bool = False
    all_domains: tuple[str, ...] = field(default_factory=lambda: DEFAULT_DOMAINS)