Skip to content

Commit 0439b4a

Browse files
committed
Freeze dask version.
1 parent 172de1e commit 0439b4a

File tree

3 files changed

+7
-7
lines changed

3 files changed

+7
-7
lines changed

python-package/xgboost/compat.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,4 +161,4 @@ def concat(value: Sequence[_T]) -> _T: # pylint: disable=too-many-return-statem
161161
d_v = arr.device.id
162162
assert d_v == d, "Concatenating arrays on different devices."
163163
return cupy.concatenate(value, axis=0)
164-
raise TypeError("Unknown type.")
164+
raise TypeError(f"Unknown type: {type(value[0])}")

tests/ci_build/conda_env/linux_sycl_test.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ dependencies:
1717
- pytest
1818
- pytest-timeout
1919
- pytest-cov
20-
- dask
20+
- dask=2024.11
2121
- dpcpp_linux-64
2222
- onedpl-devel
2323
- intel-openmp

tests/python-sycl/test_sycl_simple_dask.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,6 @@
11
from xgboost import dask as dxgb
22
from xgboost import testing as tm
33

4-
from hypothesis import given, strategies, assume, settings, note
5-
64
import dask.array as da
75
import dask.distributed
86

@@ -32,10 +30,12 @@ def test_simple(self):
3230
param["objective"] = "reg:squarederror"
3331

3432
# X and y must be Dask dataframes or arrays
35-
num_obs = 1e4
33+
num_obs = int(1e4)
3634
num_features = 20
37-
X = da.random.random(size=(num_obs, num_features), chunks=(1000, num_features))
38-
y = da.random.random(size=(num_obs, 1), chunks=(1000, 1))
35+
36+
rng = da.random.RandomState(1994)
37+
X = rng.random_sample((num_obs, num_features), chunks=(1000, -1))
38+
y = X.sum(axis=1)
3939
dtrain = dxgb.DaskDMatrix(client, X, y)
4040

4141
result = train_result(client, param, dtrain, 10)

0 commit comments

Comments
 (0)