removed sampler and some imports in the hope of fixing workflow runs

Author: SubhadityaMukherjee

openml/openml_simple.py

@@ -16,7 +16,6 @@
 from openml.extensions.sklearn import cat
 from openml.extensions.sklearn import cont
 import optuna
-import optunahub
 from sklearn.compose import ColumnTransformer
 from sklearn.ensemble import RandomForestClassifier
 from sklearn.impute import SimpleImputer
@@ -60,7 +59,6 @@
 # =====================================================
 
 ## Optuna explanation
-# Since we use `OptunaHub <https://hub.optuna.org/>`__ for the benchmarking of hyperparameter optimization,
 # we follow the `Optuna <https://github.com/optuna/optuna/>`__ search space design.
 
 ## OpenML runs
@@ -99,17 +97,10 @@ def objective(trial: optuna.Trial) -> Pipeline:
     return accuracy
 
 
-############################################################################
-# Load a sampler from OptunaHub
-# =============================
-sampler = optunahub.load_module("samplers/hebo").HEBOSampler(seed=seed)
-
-logger.log(0, f"Sampler used {sampler}")
-
 ############################################################################
 # Optimize the pipeline
 # =====================
-study = optuna.create_study(direction="maximize", sampler=sampler)
+study = optuna.create_study(direction="maximize")
 logger.log(0, f"Study {study}")
 study.optimize(objective, n_trials=15)
 

openml/requirements.txt

@@ -1,3 +1,2 @@
 openml>=0.15.1
-optunahub
-hebo
+plotly