From a396d0a9085abe9f220f6b8ae049b8c2991fdfc6 Mon Sep 17 00:00:00 2001 From: Khyathi Chandu Date: Thu, 15 Feb 2024 21:30:11 +0000 Subject: [PATCH] code qual --- analysis/per_token_reward.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/analysis/per_token_reward.py b/analysis/per_token_reward.py index aff1fc03..414cb4ce 100644 --- a/analysis/per_token_reward.py +++ b/analysis/per_token_reward.py @@ -269,7 +269,7 @@ def main(): for i, substring in enumerate(substrings): reward_so_far = round(results[i], 3) print(f"Reward: {reward_so_far} | Substring: {substring}") - tokens_per_model.append( substring) + tokens_per_model.append(substring) rewards_per_model.append(reward_so_far) tokens_list_all.append(tokens_per_model) rewards_list_all.append(rewards_per_model)