Merge pull request #271 from macrocosm-os/staging

Staging

Author: mccrindlebrian

folding/base/miner.py

@@ -53,7 +53,7 @@ def __init__(self, config=None):
             )
 
         # The axon handles request processing, allowing validators to send this miner requests.
-        self.axon = bt.axon(wallet=self.wallet, config=self.config)
+        self.axon = bt.axon(wallet=self.wallet, config=self.config, port=self.config.axon.port)
 
         # Attach determiners which functions are called when servicing a request.
         logger.info(f"Attaching forward function to miner axon.")

folding/base/validator.py

@@ -71,7 +71,7 @@ def __init__(self, config=None):
 
         # Serve axon to enable external connections.
         if not self.config.neuron.axon_off:
-            self.axon = bt.axon(wallet=self.wallet, config=self.config)
+            self.axon = bt.axon(wallet=self.wallet, config=self.config, port=self.config.axon.port)
             self._serve_axon()
 
         else:
@@ -132,17 +132,17 @@ def set_weights(self):
 
         # Calculate the average reward for each uid across non-zero values.
         # Replace any NaN values with 0.
-        raw_weights = torch.nn.functional.normalize(self.scores, p=1, dim=0)
+        raw_weights = torch.nn.functional.normalize(self.scores, p=1, dim=0).to("cpu").numpy() 
 
         logger.debug("raw_weights", raw_weights)
-        logger.debug("raw_weight_uids", self.metagraph.uids.to("cpu"))
+        logger.debug("raw_weight_uids", self.metagraph.uids)
         # Process the raw weights to final_weights via subtensor limitations.
         (
             processed_weight_uids,
             processed_weights,
         ) = bt.utils.weight_utils.process_weights_for_netuid(
-            uids=self.metagraph.uids.to("cpu"),
-            weights=raw_weights.to("cpu"),
+            uids=self.metagraph.uids,
+            weights=raw_weights,
             netuid=self.config.netuid,
             subtensor=self.subtensor,
             metagraph=self.metagraph,

folding/validators/forward.py

@@ -79,7 +79,7 @@ async def run_step(
         md_inputs=protein.md_inputs,
         pdb_contents=protein.pdb_contents,
         system_config=system_config,
-        best_submitted_energy=best_submitted_energy,
+        best_submitted_energy= 0 if np.isinf(best_submitted_energy) else best_submitted_energy,
     )
 
     # Make calls to the network with the prompt - this is synchronous.