|
8 | 8 | from typing import List
|
9 | 9 |
|
10 | 10 | from webgenie.base.neuron import BaseNeuron
|
11 |
| -from webgenie.challenges.challenge import Challenge |
| 11 | +from webgenie.challenges.challenge import Challenge, RESERVED_WEIGHTS |
12 | 12 | from webgenie.constants import CONSIDERING_SESSION_COUNTS, __STATE_VERSION__, WORK_DIR
|
13 | 13 | from webgenie.helpers.weights import save_file_to_wandb
|
14 | 14 |
|
@@ -163,21 +163,33 @@ def update_scores(self, rewards: np.ndarray, uids: List[int], challenge: Challen
|
163 | 163 |
|
164 | 164 | def get_scores(self, session_upto: int):
|
165 | 165 | scores = np.zeros(self.neuron.metagraph.n, dtype=np.float32)
|
166 |
| - tiny_weight = 1 / 128 |
167 |
| - big_weight = 1.0 |
168 | 166 | for session_number in self.session_results:
|
169 | 167 | if (session_number <= session_upto - CONSIDERING_SESSION_COUNTS or
|
170 | 168 | session_number > session_upto):
|
171 | 169 | continue
|
172 | 170 |
|
173 | 171 | winner = self.session_results[session_number]["winner"]
|
| 172 | + competition_type = self.session_results[session_number]["competition_type"] |
174 | 173 | if winner == -1:
|
175 | 174 | continue
|
176 |
| - if session_number == session_upto: |
177 |
| - scores[winner] += big_weight |
178 |
| - else: |
179 |
| - scores[winner] += tiny_weight |
| 175 | + scores[winner] += RESERVED_WEIGHTS[competition_type] |
180 | 176 | return scores
|
| 177 | + # scores = np.zeros(self.neuron.metagraph.n, dtype=np.float32) |
| 178 | + # tiny_weight = 1 / 128 |
| 179 | + # big_weight = 1.0 |
| 180 | + # for session_number in self.session_results: |
| 181 | + # if (session_number <= session_upto - CONSIDERING_SESSION_COUNTS or |
| 182 | + # session_number > session_upto): |
| 183 | + # continue |
| 184 | + |
| 185 | + # winner = self.session_results[session_number]["winner"] |
| 186 | + # if winner == -1: |
| 187 | + # continue |
| 188 | + # if session_number == session_upto: |
| 189 | + # scores[winner] += big_weight |
| 190 | + # else: |
| 191 | + # scores[winner] += tiny_weight |
| 192 | + # return scores |
181 | 193 |
|
182 | 194 | # if session_upto in self.session_results:
|
183 | 195 | # scores = self.session_results[session_upto]["scores"]
|
|
0 commit comments