fix : output trips & external education (eqasim-org#291)

Co-authored-by: Marie Laurent <[email protected]>
Symexpo · Jan 20, 2025 · 3c4dff4 · 3c4dff4
1 parent fe5b00e
commit 3c4dff4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/data/external/education.py b/data/external/education.py
@@ -16,7 +16,7 @@ def execute(context):
     ]]
 
     df_locations = df_locations[df_locations["activity_type"] == "education"]
-    df_locations = df_locations[["activity_type","education_type", "commune_id", "geometry"]].copy()
+    df_locations = df_locations[["activity_type","education_type", "commune_id","weight", "geometry"]].copy()
     df_locations["fake"] = False    
 
     df_zones = context.stage("data.spatial.municipalities")    

diff --git a/synthesis/output.py b/synthesis/output.py
@@ -183,7 +183,7 @@ def execute(context):
     if "csv" in output_formats:
         df_trips.to_csv("%s/%strips.csv" % (output_path, output_prefix), sep = ";", index = None, lineterminator = "\n")
     if "parquet" in output_formats:
-        df_trips.to_csv("%s/%strips.parquet" % (output_path, output_prefix))
+        df_trips.to_parquet("%s/%strips.parquet" % (output_path, output_prefix))
 
     # Prepare vehicles
     df_vehicle_types, df_vehicles = context.stage("synthesis.vehicles.vehicles")