permanence/reward/__init__.py

"""Reward computation utilities."""

from .engine import MAX_EPISODE_CATASTROPHE_PENALTY, REWARD_WEIGHTS, RewardBreakdown, RewardEngine
from .rubrics import (
    TaskCompletionRubric,
    PredictionAccuracyRubric,
    OptionPreservationRubric,
    CatastropheAvoidanceRubric,
    build_permanence_rubric,
)

__all__ = [
    "MAX_EPISODE_CATASTROPHE_PENALTY",
    "REWARD_WEIGHTS",
    "RewardBreakdown",
    "RewardEngine",
    "TaskCompletionRubric",
    "PredictionAccuracyRubric",
    "OptionPreservationRubric",
    "CatastropheAvoidanceRubric",
    "build_permanence_rubric",
]