NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-28 17:29:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

removing debugs

Browse source
This commit is contained in:
Shannon Sands 2025-05-18 16:48:42 -07:00
parent 24d41720ef
commit 99a64f5bce
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
atroposlib/envs/base.py
View file

@ -646,14 +646,12 @@ class BaseEnv(ABC):
group_size = group.get("group_overrides", {}).get(
"group_size", self.config.group_size
)
rprint(f"group_size: {group_size}")
rprint(f"group: {group}")
if not (
(None not in group) and (len(group.get("tokens", [])) == group_size)
):
logger.warning(
f"Group structure invalid, or token count mismatch (expected {group_size}), "
f"Got {len(group.get('tokens', []))}"
f"or 'tokens' key missing. Skipping group: {str(group)[:200]}..."
)
continue