composer_replication/trainer/__init__.py

"""composer_replication.trainer — TRL GRPOTrainer subclass + data collator.

Per docs/INTEGRATION_ARCHITECTURE.md § "Recipe A".
Per docs/adrs/ADR-003 (also wraps DiLoCo when training distributed).
"""
from __future__ import annotations

from composer_replication.trainer.composer_trainer import (
    PO_OBJECTIVES,
    ComposerReplicationTrainer,
    make_dr_grpo_config,
    make_po_config,
)

__all__ = [
    "ComposerReplicationTrainer",
    "make_dr_grpo_config",
    "make_po_config",
    "PO_OBJECTIVES",
]