code qual

allenai · Feb 15, 2024 · a396d0a · a396d0a
1 parent 3951feb
commit a396d0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)