CI friendly.

dmlc · Aug 10, 2020 · c590964 · c590964
1 parent cb538cb
commit c590964
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/common/quantile.cc b/src/common/quantile.cc
@@ -154,7 +154,7 @@ void HostSketchContainer::MakeCuts(HistogramCuts* cuts) {
     sreducer.Allreduce(dmlc::BeginPtr(reduced), nbytes, reduced.size());
   }
 
-  cuts->min_vals_.HostVector().resize(sketches_.size());
+  cuts->min_vals_.HostVector().resize(sketches_.size(), 0.0f);
   for (size_t fid = 0; fid < reduced.size(); ++fid) {
     WQSketch::SummaryContainer a;
     size_t max_num_bins = std::min(num_cuts[fid], max_bins_);
@@ -164,10 +164,10 @@ void HostSketchContainer::MakeCuts(HistogramCuts* cuts) {
       a.SetPrune(reduced[fid], max_num_bins + 1);
       CHECK(a.data && reduced[fid].data);
       const bst_float mval = a.data[0].value;
-      cuts->min_vals_.HostVector()[fid] = mval - (fabs(mval) + 1e-5f);
+      cuts->min_vals_.HostVector()[fid] = mval - fabs(mval) - 1e-5f;
     } else {
       // Empty column.
-      const float mval = kRtEps;
+      const float mval = 1e-5f;
       cuts->min_vals_.HostVector()[fid] = mval;
     }
     AddCutPoint(a, max_num_bins, cuts);

diff --git a/tests/python/test_with_dask.py b/tests/python/test_with_dask.py
@@ -483,7 +483,7 @@ def run_updater_test(self, client, params, num_rounds, dataset,
         note(history)
         assert tm.non_increasing(history['train'][dataset.metric])
 
-    @given(hist_parameter_strategy, strategies.integers(1, 20),
+    @given(hist_parameter_strategy, strategies.integers(10, 20),
            tm.dataset_strategy)
     @settings(deadline=None)
     def test_hist(self, params, num_rounds, dataset):
@@ -492,7 +492,7 @@ def test_hist(self, params, num_rounds, dataset):
                 self.run_updater_test(
                     client, params, num_rounds, dataset, 'hist')
 
-    @given(exact_parameter_strategy, strategies.integers(1, 20),
+    @given(exact_parameter_strategy, strategies.integers(10, 20),
            tm.dataset_strategy)
     @settings(deadline=None)
     def test_approx(self, params, num_rounds, dataset):