Fix an issue in state recovery for pg.early_stopping.StepWise.

Previously the gate predicate was applied on a Trial object, instead, it should be applied on a Measurement object.

PiperOrigin-RevId: 565092585

pyglove/ext/early_stopping/step_wise.py

@@ -96,7 +96,7 @@ def recover(self, history: Iterable[pg.tuning.Trial]):
               # For the last measurement of a pending trial, we use gate
               # predicate to determine the stopping decision.
               gate_predicate = self.plan[next_gate][1]
-              stopping_decision = gate_predicate(t, gate_history)
+              stopping_decision = gate_predicate(m, gate_history)
           gate_history.append(m)
           self._trial_gate_decision[t.id] = (next_gate, stopping_decision)
           next_gate += 1