# HG changeset patch
# User luca_milaz
# Date 1721596449 0
# Node ID be24e86b24879a759df3b888abed2943cccf7343
# Parent  b00ec73b1c575786fc47e225138b70a8cb4fef2d
Uploaded

diff -r b00ec73b1c57 -r be24e86b2487 marea_2/ras_to_bounds.py
--- a/marea_2/ras_to_bounds.py	Sun Jul 21 21:05:46 2024 +0000
+++ b/marea_2/ras_to_bounds.py	Sun Jul 21 21:14:09 2024 +0000
@@ -189,8 +189,9 @@
 
     if(ARGS.medium_selector == "Custom"):
         medium = read_dataset(ARGS.input_medium, "medium dataset")
+        medium.set_index(medium.columns[0], inplace=True)
         medium = medium.astype(float)
-        medium = medium['medium'].to_dict()
+        medium = medium[medium.columns[0]].to_dict()
     else:
         df_mediums = pd.read_csv(ARGS.tool_dir + "/local/medium/medium.csv", index_col = 0)
         ARGS.medium_selector = ARGS.medium_selector.replace("_", " ")