Add data split mode to DMatrix MetaInfo (#8568)

2022-12-25 04:37:37 -08:00
parent 77b069c25d
commit 3ceeb8c61c
20 changed files with 113 additions and 103 deletions
--- a/tests/cpp/helpers.cc
+++ b/tests/cpp/helpers.cc
@@ -527,8 +527,7 @@ std::unique_ptr<DMatrix> CreateSparsePageDMatrixWithRC(
  if (page_size > 0) {
    uri += "#" + tmp_file + ".cache";
  }
-  std::unique_ptr<DMatrix> dmat(
-      DMatrix::Load(uri, true, DataSplitMode::kNone, "auto"));
+  std::unique_ptr<DMatrix> dmat(DMatrix::Load(uri));
  return dmat;
 }