Skip to content

Commit 9dc64a3

Browse files
committed
FIX change assert_warns import
1 parent 411b868 commit 9dc64a3

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

sklearn/utils/tests/test_validation.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,14 @@
66
from itertools import product
77

88
import numpy as np
9-
from numpy.testing import assert_array_equal, assert_warns
9+
from numpy.testing import assert_array_equal
1010
import scipy.sparse as sp
1111
from nose.tools import assert_raises, assert_true, assert_false, assert_equal
1212

1313
from sklearn.utils.testing import assert_raises_regexp
1414
from sklearn.utils.testing import assert_no_warnings
1515
from sklearn.utils.testing import assert_warns_message
16+
from sklearn.utils.testing import assert_warns
1617
from sklearn.utils import as_float_array, check_array, check_symmetric
1718
from sklearn.utils import check_X_y
1819
from sklearn.utils.mocking import MockDataFrame
@@ -259,6 +260,7 @@ def test_check_array_dtype_warning():
259260
X_checked = assert_warns(DataConversionWarning, check_array, X,
260261
dtype=np.float64,
261262
accept_sparse=True, warn_on_dtype=True)
263+
print(X_checked)
262264
assert_equal(X_checked.dtype, np.float64)
263265

264266
# Check that the warning message includes the name of the Estimator

0 commit comments

Comments
 (0)