diff --git a/aiaccel/hpo/samplers/nelder_mead_sampler.py b/aiaccel/hpo/samplers/nelder_mead_sampler.py index 3e4aacb62..d5774545f 100644 --- a/aiaccel/hpo/samplers/nelder_mead_sampler.py +++ b/aiaccel/hpo/samplers/nelder_mead_sampler.py @@ -53,7 +53,10 @@ def before_trial(self, study: Study, trial: FrozenTrial) -> None: if "fixed_params" in trial.system_attrs: # enqueued trial self.nm.enqueued() fixed_params = trial.system_attrs["fixed_params"] - params = np.array([fixed_params[name] for name in self._search_space]) + if fixed_params.keys() == self._search_space.keys(): + params = np.array([fixed_params[name] for name in self._search_space]) + else: + raise RuntimeError("All parameters must be given when executing enqueue_trial.") else: while True: params = self.nm.get_vertex()