Limit rounds.

dmlc · Oct 3, 2020 · 29f4348 · 29f4348
1 parent bebf445
commit 29f4348
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python-gpu/test_gpu_updaters.py b/tests/python-gpu/test_gpu_updaters.py
@@ -65,7 +65,7 @@ def run_categorical_basic(self, cat, onehot, label, rounds):
         assert tm.non_increasing(by_builtin_results['Train']['rmse'])
 
     @given(strategies.integers(10, 400), strategies.integers(5, 10),
-           strategies.integers(1, 6), strategies.integers(4, 8))
+           strategies.integers(1, 5), strategies.integers(4, 8))
     @settings(deadline=None)
     @pytest.mark.skipif(**tm.no_pandas())
     def test_categorical(self, rows, cols, rounds, cats):