min max.

dmlc · Oct 22, 2024 · 763a286 · 763a286
1 parent 72b5fea
commit 763a286
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-package/xgboost/testing/updater.py b/python-package/xgboost/testing/updater.py
@@ -427,7 +427,7 @@ def _create_dmatrix(  # pylint: disable=too-many-arguments
     extmem: bool,
     enable_categorical: bool,
 ) -> xgb.DMatrix:
-    n_batches = min(2, n_samples)
+    n_batches = max(min(2, n_samples), 1)
     it = CatIter(
         n_samples // n_batches,
         n_features,