From 4056974e371439280f3e74b875dd80f5aafb1cc9 Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Mon, 26 Sep 2022 22:22:11 +0800
Subject: [PATCH] Fix sparse threshold warning. (#8268)

---
 src/data/iterative_dmatrix.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/data/iterative_dmatrix.cc b/src/data/iterative_dmatrix.cc
index 30583a943..83287dccf 100644
--- a/src/data/iterative_dmatrix.cc
+++ b/src/data/iterative_dmatrix.cc
@@ -244,7 +244,8 @@ BatchSet<GHistIndexMatrix> IterativeDMatrix::GetGradientIndex(BatchParam const&
     ghist_ = std::make_shared<GHistIndexMatrix>(&ctx_, Info(), *ellpack_, param);
   }
 
-  if (param.sparse_thresh != tree::TrainParam::DftSparseThreshold()) {
+  if (!std::isnan(param.sparse_thresh) &&
+      param.sparse_thresh != tree::TrainParam::DftSparseThreshold()) {
     LOG(WARNING) << "`sparse_threshold` can not be changed when `QuantileDMatrix` is used instead "
                     "of `DMatrix`.";
   }