Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix windows build on buildkite. (#8602)

Browse Source
This commit is contained in:
Jiaming Yuan 2022-12-16 21:12:24 +08:00 committed by GitHub
parent 43a647a4dd
commit 38887a1876
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/cpp/common/test_categorical.cc
View File

@ -66,7 +66,8 @@ TEST(Categorical, MinimalSet) {
HostDeviceVector<float> predt;
{
std::vector<float> data{kCat, kCat + 1, 32, 33, 34};
std::vector<float> data{static_cast<float>(kCat),
static_cast<float>(kCat + 1), 32.0f, 33.0f, 34.0f};
auto test = GetDMatrixFromData(data, data.size(), kCols);
learner->Predict(test, false, &predt, 0, 0, false, /*pred_leaf=*/true);
ASSERT_EQ(predt.Size(), data.size());