Skip to content

Commit eca70ee

Browse files
committed
chore: get seed from central server
1 parent 813c944 commit eca70ee

File tree

1 file changed

+17
-4
lines changed

1 file changed

+17
-4
lines changed

neurons/validators/genie_validator.py

Lines changed: 17 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -278,8 +278,17 @@ def get_seed(self, session: int, task_index: int, hash_cache: dict = {}) -> int:
278278
method = "GET"
279279
url = "http://209.126.9.130:18000/api/v1/task/seed"
280280
response = requests.request(method, url, params={"session": session, "task_number": task_index})
281-
if response.status_code == 200 and response.json()["seed"] is not None:
282-
return response.json()["seed"]
281+
if response.status_code == 200:
282+
response_json = response.json()
283+
success = response_json["success"]
284+
if not success:
285+
raise Exception(f"Failed to get seed from API: {seed}")
286+
287+
seed = response_json["seed"], response_json["task_id_seed"]
288+
289+
if seed is None:
290+
raise Exception(f"Seed is None")
291+
return seed
283292
else:
284293
raise Exception(f"Failed to get seed from API: {response.status_code}")
285294

@@ -306,15 +315,19 @@ async def forward(self):
306315
return
307316

308317
bt.logging.info(f"Forwarding task #{task_index} in session #{session}")
309-
seed = self.get_seed(session, task_index)
318+
seed, task_id_seed = self.get_seed(session, task_index)
310319

311-
bt.logging.info(f"Init random with seed: {seed}")
320+
bt.logging.info(f"Random seed: {seed} | task_id_seed: {task_id_seed}")
312321
random.seed(seed)
313322
set_seed(seed)
314323

324+
number_of_tries = 0
315325
while True:
316326
try:
317327
await self.synthensize_task()
328+
task, _ = self.synthetic_tasks[-1]
329+
number_of_tries += 1
330+
task.task_id = f"{task_id_seed}_{number_of_tries}"
318331
break
319332
except Exception as e:
320333
bt.logging.error(

0 commit comments

Comments
 (0)