mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
33 lines
856 B
Python
33 lines
856 B
Python
"""
|
|
Utility functions and classes for the atroposlib package.
|
|
"""
|
|
|
|
from .config_handler import ConfigHandler
|
|
from .message_history_utils import (
|
|
strip_thinking,
|
|
truncate_thinking,
|
|
ensure_trajectory_token_limit,
|
|
)
|
|
from .tokenize_for_trainer import tokenize_for_trainer
|
|
from .tool_call_parser import parse_tool_call
|
|
from .advantages import (
|
|
allclose_to_first,
|
|
compute_stats,
|
|
compute_discounted_returns,
|
|
compute_grpo_process_supervision_advantages,
|
|
)
|
|
from .best_of_n_selection import select_best_index
|
|
|
|
__all__ = [
|
|
"ConfigHandler",
|
|
"strip_thinking",
|
|
"truncate_thinking",
|
|
"tokenize_for_trainer",
|
|
"parse_tool_call",
|
|
"allclose_to_first",
|
|
"compute_stats",
|
|
"compute_discounted_returns",
|
|
"compute_grpo_process_supervision_advantages",
|
|
"ensure_trajectory_token_limit",
|
|
"select_best_index",
|
|
]
|