From edf9a9608e7f7db7ed722df1c0f61223df5b25d5 Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Wed, 18 May 2022 20:14:14 +0800
Subject: [PATCH] Fix type conversion warning. (#7916)

---
 src/tree/driver.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/tree/driver.h b/src/tree/driver.h
index 0aef93ccf..a4a0dd4a6 100644
--- a/src/tree/driver.h
+++ b/src/tree/driver.h
@@ -101,7 +101,7 @@ class Driver {
 
  private:
   TrainParam param_;
-  std::size_t num_leaves_ = 1;
+  bst_node_t num_leaves_ = 1;
   std::size_t max_node_batch_size_;
   ExpandQueue queue_;
 };