fix: Remove the removed MFPI_Random

automl · Sep 25, 2024 · 307fc45 · 307fc45
1 parent 5149034
commit 307fc45
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/neps/optimizers/bayesian_optimization/acquisition_functions/__init__.py b/neps/optimizers/bayesian_optimization/acquisition_functions/__init__.py
@@ -4,7 +4,6 @@
 from neps.optimizers.bayesian_optimization.acquisition_functions.ei import (
     ComprehensiveExpectedImprovement,
 )
-from neps.optimizers.bayesian_optimization.acquisition_functions.mf_pi import MFPI_Random
 from neps.optimizers.bayesian_optimization.acquisition_functions.prior_weighted import (
     DecayingPriorWeightedAcquisition,
 )
@@ -31,11 +30,6 @@
         in_fill="posterior",
         augmented_ei=True,
     ),
-    "MFPI-random": partial(
-        MFPI_Random,
-        threshold="random",
-        horizon="random",
-    ),
     "UCB": partial(
         UpperConfidenceBound,
         maximize=False,
@@ -47,5 +41,4 @@
     "ComprehensiveExpectedImprovement",
     "UpperConfidenceBound",
     "DecayingPriorWeightedAcquisition",
-    "MFPI_Random",
 ]
diff --git a/neps/optimizers/multi_fidelity/mf_bo.py b/neps/optimizers/multi_fidelity/mf_bo.py
@@ -5,7 +5,6 @@
 
 import torch
 
-from neps.optimizers.bayesian_optimization.models import SurrogateModelMapping
 from neps.optimizers.bayesian_optimization.models.ftpfn import FTPFNSurrogate
 from neps.optimizers.multi_fidelity.utils import (
     MFObservedData,