Fix empty weight.

dmlc · Jul 6, 2021 · ba3b228 · ba3b228
1 parent 0e2d5f8
commit ba3b228
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/common/quantile.cc b/src/common/quantile.cc
@@ -120,6 +120,10 @@ std::vector<float> MergeWeights(MetaInfo const &info,
 
 std::vector<float> UnrollGroupWeights(MetaInfo const &info) {
   std::vector<float> const &group_weights = info.weights_.HostVector();
+  if (group_weights.empty()) {
+    return group_weights;
+  }
+
   size_t n_samples = info.num_row_;
   auto const &group_ptr = info.group_ptr_;
   std::vector<float> results(n_samples);