Update boruta_py.py

scikit-learn-contrib · Jan 31, 2019 · 06febb1 · 06febb1
1 parent b549626
commit 06febb1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/boruta/boruta_py.py b/boruta/boruta_py.py
@@ -416,7 +416,7 @@ def _add_shadows_get_imps(self, X, y, dec_reg):
     def _assign_hits(self, hit_reg, cur_imp, imp_sha_max):
         # register hits for features that did better than the best of shadows
         cur_imp_no_nan = cur_imp[0]
-        cur_imp_no_nan[np.isnan(cur_imp_no_nan)]=0
+        cur_imp_no_nan[np.isnan(cur_imp_no_nan)] = 0
         hits = np.where(cur_imp_no_nan > imp_sha_max)[0]
         hit_reg[hits] += 1
         return hit_reg