Fix failing test

jinlow · Sep 18, 2023 · b0d9c06 · b0d9c06
1 parent 99cb8e8
commit b0d9c06
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/py-forust/tests/test_booster.py b/py-forust/tests/test_booster.py
@@ -450,6 +450,7 @@ def test_booster_to_xgboosts_with_contributions_missing_branch_methods(X_y):
         nbins=500,
         parallel=True,
         base_score=0.5,
+        missing_node_treatment="AssignToParent",
     )
     fmod.fit(X, y=y)
     fmod_preds = fmod.predict(X)