NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

new env runs locally

Browse source
This commit is contained in:
Shannon Sands 2025-05-14 11:57:45 -07:00
parent 54ae40840d
commit d6f9d58606
4 changed files with 204 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

28
atroposlib/utils/__init__.py
View file

@ -3,31 +3,5 @@ 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",
]
__all__ = ["ConfigHandler"]