Fix best_cycles update

app.py

@@ -482,11 +482,11 @@ def perf_takehome_reward_fn(completions, prompts=None, **kwargs):
                 reward += EXEC_REWARD
         if code.strip() in EXAMPLE_CODE_SET:
             reward = max(0.0, reward - COPY_PENALTY)
-            cycles = result.get("cycles")
-            with state_lock:
-                if isinstance(cycles, int) and cycles < training_state["best_cycles"]:
-                    training_state["best_cycles"] = cycles
-                    training_state["best_code"] = code
+        cycles = result.get("cycles")
+        with state_lock:
+            if isinstance(cycles, int) and cycles < training_state["best_cycles"]:
+                training_state["best_cycles"] = cycles
+                training_state["best_code"] = code
         rewards.append(float(reward))
     return rewards