Drop superfluous column prior to plotting

nismod · Nov 30, 2023 · ce5554b · ce5554b
1 parent d97d2d6
commit ce5554b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/workflow/scripts/exposure/plot_exposure_distributions.py b/workflow/scripts/exposure/plot_exposure_distributions.py
@@ -92,7 +92,7 @@ def plot_event_distributions(
     exposure_all_storms = pd.read_parquet(snakemake.input.exposure_by_event)
 
     logging.info("Aggregating (sum exposure) by event")
-    per_event = exposure_all_storms.groupby("event_id").sum()
+    per_event = exposure_all_storms.drop(columns=["edge"]).groupby("event_id").sum()
 
     logging.info("Plotting event distributions")
     os.makedirs(snakemake.output.country_event_distributions)