fix lower case culture

retrieve_coefs_max_yield.py

@@ -18,10 +18,10 @@ def get_coefs_Kc_Ky_and_max_yield(culture: str, region: str = "Bourgogne-Franche
     #CHECK THE REGION AND CROP
     if region not in ["Bourgogne-Franche-Comté"]:
         raise ValueError("The data region is not available.")
-    if culture not in ["Colza d'hiver", "Orge d'hiver", "Blé tendre d'hiver"]:
+    if culture.lower() not in ["colza d'hiver", "orge d'hiver", "blé tendre d'hiver"]:
         raise ValueError("The crop is not available.")
 
-    if culture=="Colza d'hiver":
+    if culture.lower()=="colza d'hiver":
         culture_yield = "Colza grain d'hiver"
     else:
         culture_yield = culture
@@ -31,8 +31,9 @@ def get_coefs_Kc_Ky_and_max_yield(culture: str, region: str = "Bourgogne-Franche
     df_coef = pd.read_csv("data/data_yield/Coefs_yield_franche_comte.csv", sep =";")
     # filter the region
     df_coef = df_coef[df_coef["Région"]==region]
+    df_coef["Culture"] = df_coef["Culture"].str.lower()
     # filter the crop
-    df_coef = df_coef[df_coef["Culture"]==culture]
+    df_coef = df_coef[df_coef["Culture"]==culture.lower()]
     # drop the columns "Stade de développement"
     df_coef = df_coef.drop(columns=["Stade de développement","Région","Culture"])
     
@@ -57,7 +58,7 @@ if __name__ == "__main__":
 
     # Exemple d'utilisation
     region = "Bourgogne-Franche-Comté"
-    culture = "Blé tendre d'hiver"
+    culture = "blé tendre d'hiver"
     df_coef, yield_max = get_coefs_Kc_Ky_and_max_yield(culture, region)
     
     print(df_coef)