Update feature_handler.py

feature_handler.py

@@ -1,6 +1,6 @@
 import pandas as pd
 import numpy as np
-
+from config import CITIES_COORDS
 def apply_feature_engineering(df_input):
     df = df_input.copy()
     df['time'] = pd.to_datetime(df['time'])
@@ -35,10 +35,12 @@ def apply_feature_engineering(df_input):
     df['is_hot_day'] = (df['temperature_2m_max'] >= 35).astype(int)
     df['is_heavy_rain'] = (df['precipitation_sum'] >= 20).astype(int)
     
-    # 4. Gestion de la Région (Le preprocessor attend une colonne 'region')
-    # On va chercher la région dans ton dictionnaire CITIES_COORDS
     def get_region(city):
-        return CITIES_COORDS.get(city, {}).get('region', 'Littoral')
+        city_info = CITIES_COORDS.get(city)
+        if city_info:
+            return city_info.get('region', 'Littoral')
+        return 'Littoral'
+
     df['region'] = df['city'].apply(get_region)
 
     # 5. Séries temporelles (Lags)