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:13:21 +00:00
parent 551cc7187d
commit dad40fba6b
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

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