Decrease size of trees in test

microsoft · Sep 21, 2024 · 9e3e8ed · 9e3e8ed
1 parent 3fe4577
commit 9e3e8ed
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/python_package_test/test_dask.py b/tests/python_package_test/test_dask.py
@@ -1450,12 +1450,11 @@ def test_init_score(task, output, cluster):
         _, _, _, _, dX, dy, dw, dg = _create_data(objective=task, output=output, group=None)
 
         model_factory = task_to_dask_factory[task]
+        rnd = np.random.RandomState(42)
 
-        params = {"n_estimators": 30, "num_leaves": 15, "time_out": 5}
+        params = {"n_estimators": 1, "num_leaves": 2, "time_out": 5}
         num_classes = 3 if task == "multiclass-classification" else 1
 
-        rnd = np.random.RandomState(42)
-
         if output.startswith("dataframe"):
             init_scores = dy.map_partitions(lambda x: pd.DataFrame(rnd.uniform(size=(x.size, num_classes))))
         else: