correct name

landoskape · May 10, 2024 · dfab6aa · dfab6aa
1 parent 4790dc9
commit dfab6aa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dominoes/datasets/base.py b/dominoes/datasets/base.py
@@ -257,9 +257,9 @@ def get_choice_score(self, choices, scores):
         """
         return torch.gather(scores, 2, choices.unsqueeze(2)).squeeze(2)
 
-    def process_reward(self, rewards, scores, choices, gamma_transform):
+    def process_rewards(self, rewards, scores, choices, gamma_transform):
         """
-        process the reward for performing policy gradient
+        process the rewards for performing policy gradient
 
         args:
             rewards: list of torch.Tensor, the rewards for each network (precomputed using `reward_function`)