From 0f0369d1466b8db65fd7b4fe3b44aead8ebc6f61 Mon Sep 17 00:00:00 2001
From: fis <jm.yuan@outlook.com>
Date: Thu, 29 Apr 2021 18:28:14 +0800
Subject: [PATCH] Fix approx.

---
 tests/python/test_with_dask.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/python/test_with_dask.py b/tests/python/test_with_dask.py
index 3ba9659ebb56..4818e08dc546 100644
--- a/tests/python/test_with_dask.py
+++ b/tests/python/test_with_dask.py
@@ -1027,6 +1027,9 @@ def run_updater_test(
         def is_stump():
             return params["max_depth"] == 1 or params["max_leaves"] == 1
 
+        def minimum_bin():
+            return "max_bin" in params and params["max_bin"] == 2
+
         if params["max_bin"] == 2 and is_stump():
             assert tm.non_increasing(history, tolerance=1e-3)
         else: