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

fix: align ScamOrRugEnv with BaseEnv API and add wandb logging

Browse source
This commit is contained in:
kokoron 2026-03-31 10:15:11 +00:00
parent dad40fba6b
commit 4c0c2b7e8b
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
environments/community/scam_or_rug_onchain/scam_or_rug_env.py
View file

@ -300,8 +300,8 @@ class ScamOrRugEnv(BaseEnv):
add_generation_prompt=True,
)
tokens = full_text
prompt_len = len(prompt_text)
masks = [-100] * prompt_len + [1] * (len(full_text) - prompt_len)
prompt_len = len(prompt_text)
masks = [-100] * prompt_len + [1] * (len(full_text) - prompt_len)
scored["tokens"].append(tokens)
scored["masks"].append(masks)