Skip to content

Commit

Permalink
import all_estimators from sklearn.utils
Browse files Browse the repository at this point in the history
  • Loading branch information
thierrymoudiki committed Jun 7, 2024
1 parent 05c711b commit 29a8b48
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 2 deletions.
5 changes: 4 additions & 1 deletion GPopt/GPOpt.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,10 @@
import shelve
from collections import namedtuple
from functools import partial
from sklearn.utils.discovery import all_estimators
try:
from sklearn.utils.discovery import all_estimators
except:
from sklearn.utils import all_estimators
from sklearn.base import RegressorMixin
from sklearn.gaussian_process import GaussianProcessRegressor
from sklearn.ensemble import RandomForestRegressor
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from codecs import open
from os import path

__version__ = "0.6.0"
__version__ = "0.6.1"

subprocess.call("pip install -r requirements.txt", shell=True)

Expand Down

0 comments on commit 29a8b48

Please sign in to comment.