From f04c30c479d1cbfc35fe4fe41a4fdcf788947bc6 Mon Sep 17 00:00:00 2001
From: sora <moussa.sissoko6@etu.univ-lorraine.fr>
Date: Tue, 7 Jul 2026 14:40:23 +0000
Subject: [PATCH] feat: Add management of  feat cols

---
 src/models/ensemble.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/models/ensemble.py b/src/models/ensemble.py
index cf1df63..adf270f 100644
--- a/src/models/ensemble.py
+++ b/src/models/ensemble.py
@@ -45,6 +45,10 @@ def _available(df: pd.DataFrame) -> List[str]:
 
 
 def _prepare_X(df: pd.DataFrame, features: List[str]) -> pd.DataFrame:
+    df_prepared = df.copy()
+    for col in features:
+        if col not in df_prepared.columns:
+            df_prepared[col] = 0.0
     return df[features].fillna(0).replace([np.inf, -np.inf], 0)