Merge pull request #86 from aleksandrkaekhtin/feature/rewards

Fix rewards distribution
ton-society · Jul 12, 2024 · 8e8ed9d · 8e8ed9d
2 parents 2b37d02 + 7cca134
commit 8e8ed9d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/models/scores.py b/models/scores.py
@@ -81,6 +81,7 @@ def normalized_min_max(self, p: ProjectStat, field: str, metrics):
     """
     def calculate_rewards(self, projects: List[ProjectStat]) -> List[ProjectStat]:
         remains = 0
+        assigned_full_rewards = 0
         if not self.reward_list:
             return projects
         reward_iterator = iter(self.reward_list)
@@ -92,13 +93,14 @@ def calculate_rewards(self, projects: List[ProjectStat]) -> List[ProjectStat]:
                     remains += reward - project.metrics[ProjectStat.POSSIBLE_REWARD]
                 else:
                     project.metrics[ProjectStat.REWARD] = reward
+                    if not project.metrics[ProjectStat.POSSIBLE_REWARD]:
+                        assigned_full_rewards += reward
             else:
                 project.metrics[ProjectStat.REWARD] = 0
         if remains:
-            assigned_rewards = sum(self.reward_list) - remains
             for project in projects:
                 if project.metrics[ProjectStat.REWARD] and not project.metrics[ProjectStat.POSSIBLE_REWARD]:
-                    project.metrics[ProjectStat.REWARD] += remains * project.metrics[ProjectStat.REWARD] / assigned_rewards
+                    project.metrics[ProjectStat.REWARD] += remains * project.metrics[ProjectStat.REWARD] / assigned_full_rewards
         return projects
 
     def calculate(self, metrics: List[ProjectStat]):