fix visualize to adapt to forecast.py

visualize/visualize.py

@@ -197,36 +197,10 @@ def get_plots():
             "index": "time",
         }
     )
-    folder_to_parse_moderate = "climate_data_moderate/"
-    climate_sub_folder_moderate = [
-        os.path.join(folder_to_parse_moderate, e)
-        for e in os.listdir(folder_to_parse_moderate)
-        if os.path.isdir(os.path.join(folder_to_parse_moderate, e))
-    ]
-    climate_sub_files_moderate = [
-        os.path.join(e, i)
-        for e in climate_sub_folder_moderate
-        for i in os.listdir(e)
-        if i.endswith(".nc")
-    ]
-    datasets_moderate = get_forecast_datasets(climate_sub_files_moderate)
-    moderate = get_forecast_data(datasets_moderate, latitude, longitude)
 
-    folder_to_parse_pessimist = "climate_data_pessimist/"
-    climate_sub_folder_pessimist = [
-        os.path.join(folder_to_parse_pessimist, e)
-        for e in os.listdir(folder_to_parse_pessimist)
-        if os.path.isdir(os.path.join(folder_to_parse_pessimist, e))
-    ]
-    climate_sub_files_pessimist = [
-        os.path.join(e, i)
-        for e in climate_sub_folder_pessimist
-        for i in os.listdir(e)
-        if i.endswith(".nc")
-    ]
-    datasets_pessimist = get_forecast_datasets(climate_sub_files_pessimist)
-    pessimist = get_forecast_data(datasets_pessimist, latitude, longitude)
-    moderate = moderate.rename(columns={"Precipitation (kg m-2 s-1)": "Precipitation (mm)"})#TODO : do yearly precipitations
+    moderate = get_forecast_data(latitude, longitude, "moderate")
+    pessimist = get_forecast_data(latitude, longitude, "pessimist")
+    moderate = moderate.rename(columns={"Precipitation (kg m-2 s-1)": "Precipitation (mm)"})
     moderate["time"] = pd.to_datetime(moderate["time"])
     moderate = moderate.sort_values('time')
     moderate['year'] = moderate["time"].dt.year