Skip to content

Commit 7cfea35

Browse files
committed
Change dtype check conditional to look for everything other than int
1 parent a5acf9d commit 7cfea35

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

sklearn/utils/random.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -232,8 +232,9 @@ def random_choice_csc(n_samples, classes, class_probability=None,
232232

233233
for j in range(len(classes)):
234234
classes[j] = np.asarray(classes[j])
235-
if classes[j].dtype.kind is 'f':
236-
raise ValueError("Class type float is not supported")
235+
if classes[j].dtype.kind != 'i':
236+
raise ValueError("class dtype %s is not supported" %
237+
classes[j].dtype)
237238
classes[j] = classes[j].astype(int)
238239

239240
# use uniform distribution if no class_probability is given

0 commit comments

Comments
 (0)