Skip to content

Commit 3990e13

Browse files
committed
ENH catch expected warning in ward clustering
1 parent d8be1c3 commit 3990e13

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sklearn/cluster/tests/test_hierarchical.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,8 @@ def test_connectivity_fixing_non_lil():
184184
m = np.array([[True, False], [False, True]])
185185
c = grid_to_graph(n_x=2, n_y=2, mask=m)
186186
w = Ward(connectivity=c)
187-
w.fit(x)
187+
with warnings.catch_warnings(record=True):
188+
w.fit(x)
188189

189190

190191
if __name__ == '__main__':

0 commit comments

Comments
 (0)