Skip to content

Commit 024bd7f

Browse files
authored
Merge pull request #3344 from lanzagar/sklimpute
preprocess: Update deprecated sklearn import (Imputer->SimpleImputer)
2 parents 23e622a + 0b30028 commit 024bd7f

File tree

2 files changed

+9
-11
lines changed

2 files changed

+9
-11
lines changed

Orange/preprocess/preprocess.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,10 @@
33
----------
44
55
"""
6+
import bottleneck as bn
67
import numpy as np
78
import scipy.sparse as sp
8-
import sklearn.preprocessing as skl_preprocessing
9-
import bottleneck as bn
9+
from sklearn.impute import SimpleImputer
1010

1111
import Orange.data
1212
from Orange.data.filter import HasClass
@@ -147,7 +147,7 @@ def __call__(self, data):
147147

148148

149149
class SklImpute(Preprocess):
150-
__wraps__ = skl_preprocessing.Imputer
150+
__wraps__ = SimpleImputer
151151

152152
def __init__(self, strategy='mean'):
153153
self.strategy = strategy
@@ -156,7 +156,7 @@ def __call__(self, data):
156156
from Orange.data.sql.table import SqlTable
157157
if isinstance(data, SqlTable):
158158
return Impute()(data)
159-
imputer = skl_preprocessing.Imputer(strategy=self.strategy)
159+
imputer = SimpleImputer(strategy=self.strategy)
160160
X = imputer.fit_transform(data.X)
161161
# Create new variables with appropriate `compute_value`, but
162162
# drop the ones which do not have valid `imputer.statistics_`

Orange/widgets/tests/base.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,6 @@
4040
ANNOTATED_DATA_FEATURE_NAME, ANNOTATED_DATA_SIGNAL_NAME
4141
)
4242
from Orange.widgets.utils.owlearnerwidget import OWBaseLearner
43-
from Orange.widgets.utils.plot import OWPlotGUI
4443
from Orange.widgets.widget import OWWidget
4544

4645
sip.setdestroyonexit(False)
@@ -146,27 +145,27 @@ def create_widget(self, cls, stored_settings=None, reset_default_settings=True):
146145
return widget
147146

148147
@staticmethod
149-
def reset_default_settings(cls):
148+
def reset_default_settings(widget):
150149
"""Reset default setting values for widget
151150
152151
Discards settings read from disk and changes stored by fast_save
153152
154153
Parameters
155154
----------
156-
cls : OWWidget
155+
widget : OWWidget
157156
widget to reset settings for
158157
"""
159-
settings_handler = getattr(cls, "settingsHandler", None)
158+
settings_handler = getattr(widget, "settingsHandler", None)
160159
if settings_handler:
161160
# Rebind settings handler to get fresh copies of settings
162161
# in known_settings
163-
settings_handler.bind(cls)
162+
settings_handler.bind(widget)
164163
# Reset defaults read from disk
165164
settings_handler.defaults = {}
166165
# Reset context settings
167166
settings_handler.global_contexts = []
168167

169-
def process_events(self, until: callable=None, timeout=DEFAULT_TIMEOUT):
168+
def process_events(self, until: callable = None, timeout=DEFAULT_TIMEOUT):
170169
"""Process Qt events, optionally until `until` returns
171170
something True-ish.
172171
@@ -1084,6 +1083,5 @@ def datasets(cls):
10841083
yield cls.missing_data_2()
10851084
yield cls.missing_data_3()
10861085
yield cls.data_one_column_nans()
1087-
yield cls.data_one_column_infs()
10881086
yield ds_cls
10891087
yield ds_reg

0 commit comments

Comments
 (0)