chore: Remove emukit

PiperOrigin-RevId: 674428570
google · Sep 13, 2024 · cdca538 · cdca538
1 parent a0b6f82
commit cdca538
Show file tree

Hide file tree

Showing 7 changed files with 0 additions and 304 deletions.
diff --git a/requirements-algorithms.txt b/requirements-algorithms.txt
@@ -1,7 +1,3 @@
-# Emukit BayesOpt library.
-emukit==0.4.9
-scipy>1.2.3,<1.8.0 # Required downgrade to allow emukit batching behavior.
-
 # For BOCS.
 cvxpy==1.2.1
 cvxopt==1.3.0

diff --git a/vizier/_src/algorithms/designers/emukit.py b/vizier/_src/algorithms/designers/emukit.py
diff --git a/vizier/_src/algorithms/designers/emukit_test.py b/vizier/_src/algorithms/designers/emukit_test.py
diff --git a/vizier/_src/pyglove/algorithms.py b/vizier/_src/pyglove/algorithms.py
@@ -55,7 +55,6 @@ def _should_never_trigger(self, method_name):
  # OSS-Vizier only algorithms:
  'BOCS',
  'CMA_ES',
- 'EMUKIT_GP_EI',
  'HARMONICA',
  'NSGA2',
  'QUASI_RANDOM_SEARCH',

diff --git a/vizier/_src/pyvizier/oss/study_config.py b/vizier/_src/pyvizier/oss/study_config.py
@@ -78,8 +78,6 @@ class Algorithm(enum.Enum):
  QUASI_RANDOM_SEARCH = 'QUASI_RANDOM_SEARCH'
  # NSGA2 (https://ieeexplore.ieee.org/document/996017).
  NSGA2 = 'NSGA2'
- # Emukit implementation of GP-EI (https://emukit.github.io/).
- EMUKIT_GP_EI = 'EMUKIT_GP_EI'
  # BOCS (https://arxiv.org/abs/1806.08838) only applicable to boolean search
  # spaces.
  BOCS = 'BOCS'

diff --git a/vizier/_src/service/policy_factory.py b/vizier/_src/service/policy_factory.py
@@ -89,10 +89,6 @@ def __call__(
  return dp.PartiallySerializableDesignerPolicy(
  problem_statement, policy_supporter, nsga2.NSGA2Designer
  )
- elif algorithm == 'EMUKIT_GP_EI':
- from vizier._src.algorithms.designers import emukit
-
- return dp.DesignerPolicy(policy_supporter, emukit.EmukitDesigner)
  elif algorithm == 'BOCS':
  from vizier._src.algorithms.designers import bocs
 

diff --git a/vizier/algorithms/designers/__init__.py b/vizier/algorithms/designers/__init__.py
@@ -19,7 +19,6 @@
 from vizier._src.algorithms.designers.bocs import BOCSDesigner
 from vizier._src.algorithms.designers.cmaes import CMAESDesigner
 from vizier._src.algorithms.designers.eagle_strategy.eagle_strategy import EagleStrategyDesigner
-from vizier._src.algorithms.designers.emukit import EmukitDesigner
 from vizier._src.algorithms.designers.gp_bandit import VizierGPBandit
 from vizier._src.algorithms.designers.gp_ucb_pe import VizierGPUCBPEBandit
 from vizier._src.algorithms.designers.grid import GridSearchDesigner