External memory test.

dmlc · Nov 27, 2021 · dc6359e · dc6359e
1 parent 904b787
commit dc6359e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/test_data_iterator.py b/tests/python/test_data_iterator.py
@@ -111,7 +111,7 @@ def run_data_iterator(
     )
     arr_predt = from_arrays.predict(Xy)
 
-    if tree_method != "gpu_hist":
+    if tree_method == "hist":
         rtol = 1e-1  # flaky
     else:
         # Model can be sensitive to quantiles, use 1e-2 to relax the test.