From 4c0c2b7e8bdf0e7fd81a0253ad766958f1e96b08 Mon Sep 17 00:00:00 2001 From: kokoron <78990112+ILKokoron@users.noreply.github.com> Date: Tue, 31 Mar 2026 10:15:11 +0000 Subject: [PATCH] fix: align ScamOrRugEnv with BaseEnv API and add wandb logging --- environments/community/scam_or_rug_onchain/scam_or_rug_env.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/environments/community/scam_or_rug_onchain/scam_or_rug_env.py b/environments/community/scam_or_rug_onchain/scam_or_rug_env.py index a5451dfa..f57c37da 100644 --- a/environments/community/scam_or_rug_onchain/scam_or_rug_env.py +++ b/environments/community/scam_or_rug_onchain/scam_or_rug_env.py @@ -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)