8000 FIX Explicitly ignore SparseEfficiencyWarning in DBSCAN (#13539) · scikit-learn/scikit-learn@6216b24 · GitHub
[go: up one dir, main page]
Skip to content

Commit 6216b24

Browse files
peayTomDLT
peay
authored and
TomDLT
committed
FIX Explicitly ignore SparseEfficiencyWarning in DBSCAN (#13539)
1 parent 26f6909 commit 6216b24

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

‎sklearn/cluster/dbscan_.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,11 @@
1010
# License: BSD 3 clause
1111

1212
import numpy as np
13+
import warnings
1314
from scipy import sparse
1415

1516
from ..base import BaseEstimator, ClusterMixin
1617
from ..utils import check_array, check_consistent_length
17-
from ..utils.testing import ignore_warnings
1818
from ..neighbors import NearestNeighbors
1919

2020
from ._dbscan_inner import dbscan_inner
@@ -145,7 +145,8 @@ def dbscan(X, eps=0.5, min_samples=5, metric='minkowski', metric_params=None,
145145
X.sum_duplicates() # XXX: modifies X's internals in-place
146146

147147
# set the diagonal to explicit values, as a point is its own neighbor
148-
with ignore_warnings():
148+
with warnings.catch_warnings():
149+
warnings.simplefilter('ignore', sparse.SparseEfficiencyWarning)
149150
X.setdiag(X.diagonal()) # XXX: modifies X's internals in-place
150151

151152
X_mask = X.data <= eps

0 commit comments

Comments
 (0)
0