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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-02-21 19:25:11 +00:00
parent 7e5ddbce06
commit a930d3db12
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
environments/game_environments/gymnasium/blackjack/blackjack_env_no_thinking.py
View file

@ -181,7 +181,9 @@ class BlackjackEnvNoThinking(BaseEnv):
async with self.server.dedicated_server() as server:
for _ in range(self.config.max_episode_turns):
if (
len(self.tokenizer.apply_chat_template(messages, tokenize=False))
len(
self.tokenizer.apply_chat_template(messages, tokenize=False)
)
> self.config.max_token_length - 50
):
logger.warning(
@ -208,7 +210,9 @@ class BlackjackEnvNoThinking(BaseEnv):
logger.error(f"[Seed: {seed}] LLM API error: {e}")
break
messages.append({"role": "assistant", "content": llm_action_response})
messages.append(
{"role": "assistant", "content": llm_action_response}
)
action = self._parse_action_from_llm(llm_action_response)
if action is None: