revert change to objective

unit8co · Nov 3, 2023 · 3aab3b1 · 3aab3b1
1 parent 2bd75e9
commit 3aab3b1
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/darts/models/forecasting/xgboost.py b/darts/models/forecasting/xgboost.py
@@ -266,9 +266,8 @@ def fit(
                 if xgb_200_or_above:
                     self.kwargs["quantile_alpha"] = quantile
                 else:
-                    self.kwargs["objective"] = partial(
-                        xgb_quantile_loss, quantile=quantile
-                    )
+                    objective = partial(xgb_quantile_loss, quantile=quantile)
+                    self.kwargs["objective"] = objective
                 self.model = xgb.XGBRegressor(**self.kwargs)
 
                 super().fit(