update test

RektPunk · Oct 5, 2024 · eb5cc98 · eb5cc98
1 parent c730b73
commit eb5cc98
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_engine.py b/tests/test_engine.py
@@ -66,6 +66,7 @@ def test_rektengine_xgb_fit(mock_dataset, mock_xgb_model):
     xgb.train.assert_called_once_with(
         dtrain="dtrain_mock",
         verbose_eval=False,
+        num_boost_round=100,
         params={"eval_metric": "rmse"},
         evals_result={},
         evals=[("dvalid_mock", _VALID_STR)],