Skip to content

Commit

Permalink
New space initialization
Browse files Browse the repository at this point in the history
  • Loading branch information
wagnersj committed Mar 31, 2022
1 parent cfc54bf commit ee7139c
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 4 deletions.
2 changes: 2 additions & 0 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@

packages = [ # Packages must be sorted alphabetically to ease maintenance and merges.
"orion.algo",
"orion.algo.mofa",
"orion.algo.pbt",
"orion.analysis",
"orion.benchmark",
Expand Down Expand Up @@ -63,6 +64,7 @@
"tpe = orion.algo.tpe:TPE",
"EvolutionES = orion.algo.evolution_es:EvolutionES",
"pbt = orion.algo.pbt.pbt:PBT",
"mofa = orion.algo.mofa.mofa:MOFA",
],
"Database": [
"ephemeraldb = orion.core.io.database.ephemeraldb:EphemeralDB",
Expand Down
3 changes: 0 additions & 3 deletions src/orion/algo/mofa/mofa.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,7 +103,6 @@ def __init__(
# Initialize a dictionary for the frozen values of each hyperparameter
self.frozen_param_values = {}

def _initialize(self):
# Generate the initial set of trials
self.roi_space = self.space
self.current_trials_params = self._generate_trials_unfrozen_params(
Expand Down Expand Up @@ -189,8 +188,6 @@ def suggest(self, num):
New parameters must be compliant with the problem's domain `orion.algo.space.Space`.
"""
if not hasattr(self, "current_trials_params"):
self._initialize()

if self.n_trials is None or len(self.completed_trials) >= self.n_trials:
self._prepare_next_iteration()
Expand Down
2 changes: 1 addition & 1 deletion tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,7 @@ description = Verify code style with black
basepython = python3
skip_install = true
deps =
black == 20.8b1
black == 22.3.0
commands =
black --check src/orion/ tests/

Expand Down

0 comments on commit ee7139c

Please sign in to comment.