Skip to content

Commit e1b1200

Browse files
author
Thomas Unterthiner
committed
Make sparsefuncs_fast imports private in sparsefuncs
1 parent f441c6c commit e1b1200

1 file changed

Lines changed: 6 additions & 7 deletions

File tree

sklearn/utils/sparsefuncs.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66
import numpy as np
77

88
from .fixes import sparse_min_max
9-
from .sparsefuncs_fast import (csr_mean_variance_axis0,
10-
csc_mean_variance_axis0)
9+
from .sparsefuncs_fast import csr_mean_variance_axis0 as _csr_mean_var_axis0
10+
from .sparsefuncs_fast import csc_mean_variance_axis0 as _csc_mean_var_axis0
1111

1212

1313
def _raise_typeerror(X):
@@ -82,19 +82,18 @@ def mean_variance_axis(X, axis):
8282

8383
if isinstance(X, sp.csr_matrix):
8484
if axis == 0:
85-
return csr_mean_variance_axis0(X)
85+
return _csr_mean_var_axis0(X)
8686
else:
87-
return csc_mean_variance_axis0(X.T)
87+
return _csc_mean_var_axis0(X.T)
8888
elif isinstance(X, sp.csc_matrix):
8989
if axis == 0:
90-
return csc_mean_variance_axis0(X)
90+
return _csc_mean_var_axis0(X)
9191
else:
92-
return csr_mean_variance_axis0(X.T)
92+
return _csr_mean_var_axis0(X.T)
9393
else:
9494
_raise_typeerror(X)
9595

9696

97-
9897
def inplace_column_scale(X, scale):
9998
"""Inplace column scaling of a CSC/CSR matrix.
10099

0 commit comments

Comments
 (0)