Skip to content

Commit 8563b14

Browse files
author
Dmitry Razdoburdin
committed
fix random state for test
1 parent 3311817 commit 8563b14

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/python-sycl/test_sycl_training_continuation.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,11 @@ class TestSYCLTrainingContinuation:
99
def run_training_continuation(self, use_json):
1010
kRows = 64
1111
kCols = 32
12-
X = np.random.randn(kRows, kCols)
13-
y = np.random.randn(kRows)
12+
X = rng.randn(kRows, kCols)
13+
y = rng.randn(kRows)
1414
dtrain = xgb.DMatrix(X, y)
1515
params = {
16-
"device": "sycl",
16+
"device": "sycl:cpu",
1717
"max_depth": "2",
1818
"gamma": "0.1",
1919
"alpha": "0.01",

0 commit comments

Comments
 (0)