Merge branch 'feature/add_nm_sampler' of github:aistairc/aiaccel into…

… feature/add_nm_sampler
aistairc · Mar 7, 2024 · eeddf35 · eeddf35
2 parents 87b6272 + 3e0a517
commit eeddf35
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aiaccel/hpo/samplers/nelder_mead_sampler.py b/aiaccel/hpo/samplers/nelder_mead_sampler.py
@@ -71,10 +71,10 @@ def _generator(self) -> Generator[np.ndarray, None, None]:
         # initialization
         lows, highs = zip(*self._search_space.values())
         self.vertices = self._rng.uniform(lows, highs, (self.dimension + 1, self.dimension))
+        self.values = np.empty(self.dimension + 1)
 
         yield from self.vertices
-        results = yield from self._waiting_for_list(len(self.vertices))
-        self.values = np.array(results)
+        self.values[:] = yield from self._waiting_for_list(len(self.vertices))
 
         # main loop
         shrink_requied = False