Skip to content

Commit 4ed5ab9

Browse files
committed
Fix gtest.
1 parent deffe5d commit 4ed5ab9

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/cpp/helpers.cc

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -492,9 +492,10 @@ void MakeLabels(DeviceOrd device, bst_idx_t n_samples, bst_target_t n_classes,
492492
std::shared_ptr<DMatrix> p_fmat{
493493
DMatrix::Create(static_cast<DataIterHandle>(iter.get()), iter->Proxy(), Reset, Next, config)};
494494

495-
auto row_page_path =
496-
data::MakeId(prefix, dynamic_cast<data::SparsePageDMatrix*>(p_fmat.get())) + ".row.page";
497-
EXPECT_TRUE(FileExists(row_page_path)) << row_page_path;
495+
auto row_page_path = data::MakeId(data::MakeCachePrefix(prefix),
496+
dynamic_cast<data::SparsePageDMatrix*>(p_fmat.get())) +
497+
".row.page";
498+
EXPECT_TRUE(FileExists(row_page_path)) << row_page_path << " prefix:" << prefix;
498499

499500
// Loop over the batches and count the number of pages
500501
std::size_t batch_count = 0;

0 commit comments

Comments
 (0)