Fix typo.

aistairc · Apr 4, 2024 · a29a7ab · a29a7ab
1 parent 503322e
commit a29a7ab
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/aiaccel/hpo/samplers/nelder_mead_sampler.py b/aiaccel/hpo/samplers/nelder_mead_sampler.py
@@ -265,17 +265,15 @@ def before_trial(self, study: Study, trial: FrozenTrial) -> None:
             params = self.nm.put_enqueue_vertex_queue(trial.system_attrs["fixed_params"])
             self.enqueue_running_trial_id.append(trial._trial_id)
         else:
-            params = self.nm.get_vertex()
-
             while True:
+                params = self.nm.get_vertex()
                 if params is None:
                     raise RuntimeError("No more parallel calls to ask() are possible.")
 
                 if all(low < x < high for x, (low, high) in zip(params, self._search_space.values())):
                     break
                 else:
                     self.nm.put_value_queue(np.inf)
-                    params = self.nm.get_vertex()
             self.running_trial_id.append(trial._trial_id)
 
         trial.set_user_attr("params", params)