remove skrf

PKU-DAIR · Feb 29, 2024 · 6cbb319 · 6cbb319
1 parent 47ccb37
commit 6cbb319
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 217 deletions.
diff --git a/openbox/core/base.py b/openbox/core/base.py
@@ -74,10 +74,6 @@ def build_surrogate(func_str='gp', config_space=None, rng=None, transfer_learnin
         from openbox.surrogate.lightgbm import LightGBM
         return LightGBM(config_space, types=types, bounds=bounds, seed=seed)
 
-    if func_str == 'random_forest':
-        from openbox.surrogate.skrf import RandomForestSurrogate
-        return RandomForestSurrogate(config_space, types=types, bounds=bounds, seed=seed)
-
     elif func_str.startswith('gp'):
         from openbox.surrogate.base.build_gp import create_gp_model
         return create_gp_model(model_type=func_str,

diff --git a/openbox/surrogate/skrf.py b/openbox/surrogate/skrf.py
diff --git a/test/core/test_base.py b/test/core/test_base.py
@@ -6,7 +6,6 @@
 from openbox.surrogate.base.rf_with_instances import RandomForestWithInstances
 from openbox.surrogate.base.rf_with_instances_sklearn import skRandomForestWithInstances
 from openbox.surrogate.lightgbm import LightGBM
-from openbox.surrogate.skrf import RandomForestSurrogate
 from openbox.surrogate.base.base_gp import BaseGP
 from openbox.surrogate.tlbo.mfgpe import MFGPE
 from openbox.surrogate.tlbo.rgpe import RGPE
@@ -65,9 +64,6 @@ def test_build_surrogate(configspace_tiny, transfer_learning_history_single):
     surrogate = build_surrogate('lightgbm', config_space, rng)
     assert isinstance(surrogate, LightGBM)
 
-    surrogate = build_surrogate('random_forest', config_space, rng)
-    assert isinstance(surrogate, RandomForestSurrogate)
-
     surrogate = build_surrogate('gp', config_space, rng)
     assert isinstance(surrogate, BaseGP)
 

diff --git a/test/surrogate/test_skrf.py b/test/surrogate/test_skrf.py