Skip to content

Commit ebc1c0d

Browse files
committed
Fix nanrankdata bug when the entire row/column is NaN
1 parent 1995da7 commit ebc1c0d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

boruta/boruta_py.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -490,8 +490,8 @@ def _nanrankdata(self, X, axis=1):
490490
"""
491491
Replaces bottleneck's nanrankdata with scipy and numpy alternative.
492492
"""
493-
ranks = sp.stats.mstats.rankdata(np.ma.masked_invalid(X), axis=axis)
494-
ranks[ranks == 0] = np.nan
493+
ranks = sp.stats.mstats.rankdata(X, axis=axis)
494+
ranks[np.isnan(X)] = np.nan
495495
return ranks
496496

497497
def _check_params(self, X, y):

0 commit comments

Comments
 (0)