NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-30 17:40:36 +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-04-07 22:25:03 +00:00
parent 463aa79ae8
commit 7c67e0bb19
5 changed files with 108 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

36
atroposlib/envs/base.py
View file

@ -871,12 +871,36 @@ class BaseEnv(ABC):
metadata = {
"env": self.name,
"env_id": env_id,
"logprobs": group.get("logprobs") if group.get("logprobs") is not None else None,
"ref_logprobs": group.get("ref_logprobs") if group.get("ref_logprobs") is not None else None,
"distill_token_ids": group.get("distill_token_ids") if group.get("distill_token_ids") is not None else None,
"distill_logprobs": group.get("distill_logprobs") if group.get("distill_logprobs") is not None else None,
"overrides": group.get("overrides") if group.get("overrides") is not None else None,
"group_overrides": group.get("group_overrides") if group.get("group_overrides") is not None else None,
"logprobs": (
group.get("logprobs")
if group.get("logprobs") is not None
else None
),
"ref_logprobs": (
group.get("ref_logprobs")
if group.get("ref_logprobs") is not None
else None
),
"distill_token_ids": (
group.get("distill_token_ids")
if group.get("distill_token_ids") is not None
else None
),
"distill_logprobs": (
group.get("distill_logprobs")
if group.get("distill_logprobs") is not None
else None
),
"overrides": (
group.get("overrides")
if group.get("overrides") is not None
else None
),
"group_overrides": (
group.get("group_overrides")
if group.get("group_overrides") is not None
else None
),
}
self.shm_buffer.write_trajectory(
tokens=group["tokens"][i],