Merge pull request #363 from macrocosm-os/staging

Staging

Author: schampoux

folding/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.0.1"
+__version__ = "2.1.1"
 version_split = __version__.split(".")
 __spec_version__ = (
     (10000 * int(version_split[0]))

folding/utils/config.py

@@ -71,7 +71,7 @@ def add_args(cls, parser):
         "--neuron.epoch_length",
         type=int,
         help="The default epoch length (how often we set weights, measured in 12 second blocks).",
-        default=300,
+        default=150,
     )
 
     parser.add_argument(

folding/validators/forward.py

@@ -57,6 +57,7 @@ async def run_step(
     job_type: str,
     job_id: str,
     best_submitted_energy: float = None,
+    first: bool = False,
 ) -> Dict:
     start_time = time.time()
 
@@ -78,9 +79,9 @@ async def run_step(
     synapse = JobSubmissionSynapse(
         pdb_id=protein.pdb_id,
         job_id=job_id,
-        best_submitted_energy=0
-        if np.isinf(best_submitted_energy)
-        else 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.
@@ -101,6 +102,8 @@ async def run_step(
         "energies": [],
         **response_info,
     }
+    if first:
+        return event
 
     energies, energy_event = get_energies(
         protein=protein, responses=responses, uids=uids, job_type=job_type