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/common/test_hist_util.h
+++ b/tests/cpp/common/test_hist_util.h
@@ -88,8 +88,7 @@ inline std::shared_ptr<DMatrix> GetExternalMemoryDMatrixFromData(
    fo << row_data.str() << "\n";
  }
  fo.close();
-  return std::shared_ptr<DMatrix>(DMatrix::Load(
-      tmp_file + "#" + tmp_file + ".cache", true, DataSplitMode::kNone, "auto"));
+  return std::shared_ptr<DMatrix>(DMatrix::Load(tmp_file + "#" + tmp_file + ".cache"));
 }

 // Test that elements are approximately equally distributed among bins