From a07945cc337ceb19f7b81b6fa6467947666c27d9 Mon Sep 17 00:00:00 2001
From: Moritz Mistol <moritz.mistol@gmail.com>
Date: Wed, 8 Jan 2025 17:41:12 +0100
Subject: [PATCH] enable categorical in xgboost

that lead to a test fail when non-cyclical encodings are used
---
 src/probafcst/models/xgboost.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/probafcst/models/xgboost.py b/src/probafcst/models/xgboost.py
index 59d6dd4..25e493f 100644
--- a/src/probafcst/models/xgboost.py
+++ b/src/probafcst/models/xgboost.py
@@ -25,6 +25,7 @@ def __init__(
         model = xgb.XGBRegressor(
             objective="reg:quantileerror",
             quantile_alpha=quantiles,
+            enable_categorical=True,
             **self.kwargs,
         )
         super().__init__(