Skip to content

Commit c71253b

Browse files
author
Vladislav Nazarov
authored
Cast to float in SVC test due to fail on GPU (#868) (#876)
1 parent d4faea0 commit c71253b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onedal/svm/tests/test_svc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,8 +111,8 @@ def test_sample_weight(queue):
111111

112112
@pytest.mark.parametrize('queue', get_queues())
113113
def test_decision_function(queue):
114-
X = [[-2, -1], [-1, -1], [-1, -2], [1, 1], [1, 2], [2, 1]]
115-
Y = [1, 1, 1, 2, 2, 2]
114+
X = np.array([[-2, -1], [-1, -1], [-1, -2], [1, 1], [1, 2], [2, 1]], dtype=np.float32)
115+
Y = np.array([1, 1, 1, 2, 2, 2], dtype=np.float32)
116116

117117
clf = SVC(kernel='rbf', gamma=1, decision_function_shape='ovo')
118118
clf.fit(X, Y, queue=queue)

0 commit comments

Comments
 (0)