Skip to content

Commit fa6d659

Browse files
authored
fixing reference to self.is_lgb (#82)
There was a lost reference to self.is_lgb but now we use self._is_lightgbm
1 parent 855c4c9 commit fa6d659

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

boruta/boruta_py.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,7 @@ def _fit(self, X, y):
303303
self.estimator.set_params(n_estimators=n_tree)
304304

305305
# make sure we start with a new tree in each iteration
306-
if self.is_lgb:
306+
if self._is_lightgbm:
307307
self.estimator.set_params(random_state=self.random_state.randint(0, 10000))
308308
else:
309309
self.estimator.set_params(random_state=self.random_state)

0 commit comments

Comments
 (0)