Skip to content

Commit

Permalink
Adjust timing of _waiting_for_list.
Browse files Browse the repository at this point in the history
  • Loading branch information
KanaiYuma-aist committed Mar 22, 2024
1 parent cfd1de5 commit cc82cd1
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions aiaccel/hpo/samplers/nelder_mead_sampler.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,15 +109,13 @@ def _initialization(self) -> Generator[np.ndarray, None, None]:
vertices_of_random.append(yield_vertex)
i += 1

values_of_random, _ = yield from self._waiting_for_list(i)
values_of_random, enqueue_values = yield from self._waiting_for_list(i)

# enqueue
enqueue_vertices = []
while not self.enqueue_vertex_queue.empty():
enqueue_vertices.append(self.enqueue_vertex_queue.get(block=False))

enqueue_values = yield from self._waiting_for_enqueue_list()

self.vertices = np.array(vertices_of_random + enqueue_vertices)
self.values = np.array(values_of_random + enqueue_values)

Expand Down

0 comments on commit cc82cd1

Please sign in to comment.