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

11
environments/game_environments/gymnasium/blackjack_env_thinking.py
View file

@ -27,13 +27,10 @@ from atroposlib.envs.base import (
OpenaiConfig,
ScoredDataGroup,
)
from atroposlib.utils import (
tokenize_for_trainer,
parse_tool_call,
truncate_thinking,
ensure_trajectory_token_limit,
select_best_index
)
from atroposlib.utils.tokenize_for_trainer import tokenize_for_trainer
from atroposlib.utils.message_history_utils import truncate_thinking
from atroposlib.utils.tool_call_parser import parse_tool_call
from atroposlib.utils.best_of_n_selection import select_best_index
logger = logging.getLogger(__name__)