Merge pull request #103 from Datura-ai/hotfix-main-bittensor

wait for synthetic task complete

Author: surcyf123

validators/services/capacity.py

@@ -10,7 +10,7 @@ class CapacityService:
     def __init__(self, metagraph, dendrite):
         self.metagraph = metagraph
         self.dendrite: bt.dendrite = dendrite
-        self.timeout = 4
+        self.timeout = 30
         self.uid_to_capacity = {}
         self.remain_uid_to_capacity = {}
 

validators/weight_setter.py

@@ -348,7 +348,7 @@ async def get_available_uids(self):
     async def check_uid(self, axon, uid):
         """Asynchronously check if a UID is available."""
         try:
-            response = await self.dendrite(axon, IsAlive(), timeout=4)
+            response = await self.dendrite(axon, IsAlive(), timeout=30)
             if response.completion == 'True':
                 bt.logging.trace(f"UID {uid} is active")
                 return axon  # Return the axon info instead of the UID
@@ -599,6 +599,9 @@ async def process_queries_from_database(self):
             if not self.is_epoch_end():
                 bt.logging.debug("no end of epoch. so continue...")
                 continue
+            if not self.synthetic_task_done:
+                bt.logging.debug("wait for synthetic tasks to complete.")
+                continue
 
             bt.logging.info(f"start scoring process...")