From 876227d75cb4f19b510e6ae82cc88dd491e0483f Mon Sep 17 00:00:00 2001
From: Guillaume VIGNAL <guillaume.vignal@maif.fr>
Date: Thu, 10 Oct 2024 13:31:17 +0200
Subject: [PATCH] fix na in the dataset

---
 eurybia/core/smartdrift.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/eurybia/core/smartdrift.py b/eurybia/core/smartdrift.py
index b6d85dc..e0db28a 100644
--- a/eurybia/core/smartdrift.py
+++ b/eurybia/core/smartdrift.py
@@ -273,6 +273,7 @@ def compile(
         dtypes = df_concat[varz].dtypes.map(str)
         cat_features = list(dtypes[dtypes.isin(["object"])].index)
         df_concat[cat_features] = df_concat[cat_features].fillna("NA")
+        df_concat = df_concat.fillna(0)
         self._df_concat = df_concat
 
         train, test = train_test_split(df_concat[varz + [self._datadrift_target]], test_size=0.25, random_state=42)