Cast to float in SVC test due to fail on GPU
@@ -111,8 +111,8 @@ def test_sample_weight(queue):
@pytest.mark.parametrize('queue', get_queues())
def test_decision_function(queue):
- X = [[-2, -1], [-1, -1], [-1, -2], [1, 1], [1, 2], [2, 1]]
- Y = [1, 1, 1, 2, 2, 2]
+ X = np.array([[-2, -1], [-1, -1], [-1, -2], [1, 1], [1, 2], [2, 1]], dtype=np.float32)
+ Y = np.array([1, 1, 1, 2, 2, 2], dtype=np.float32)
clf = SVC(kernel='rbf', gamma=1, decision_function_shape='ovo')
clf.fit(X, Y, queue=queue)
