Skip to content

Commit

Permalink
Import error fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Neeratyoy committed Aug 30, 2024
1 parent e3e6889 commit 0025d3d
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions neps/optimizers/multi_fidelity/mf_bo.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,11 +9,9 @@
from neps.utils.common import instance_from_map
from neps.optimizers.bayesian_optimization.models import SurrogateModelMapping
from neps.optimizers.utils import map_real_hyperparameters_from_tabular_ids
from neps.optimizers.multi_fidelity_prior.utils import (
calc_total_resources_spent, normalize_vectorize_config, update_fidelity
)
from neps.optimizers.multi_fidelity_prior.utils import calc_total_resources_spent, update_fidelity
from neps.search_spaces.search_space import SearchSpace
from neps.optimizers.multi_fidelity.utils import
from neps.optimizers.multi_fidelity.utils import normalize_vectorize_config

class MFBOBase:
"""Designed to work with model-based search on SH-based multi-fidelity algorithms.
Expand Down Expand Up @@ -213,6 +211,10 @@ def _fantasize_pending(self, train_x, train_y, pending_x):
)
pending_condition = self.observed_configs.pending_condition
if pending_condition.any():
raise NotImplementedError(
"Fantasization not implemented yet!"
"This optimizer cannot be run with multiple workers yet."
)
print(f"\n\nFound pending: {pending_condition.sum()}\n\n")
pending_configs = self.observed_configs.df.loc[pending_condition]
self._fit(train_x, train_y, train_lcs)
Expand Down

0 comments on commit 0025d3d

Please sign in to comment.