Update scarse/functions/functions.py

scarse/functions/functions.py

@@ -151,9 +151,9 @@ class ModelOptimization:
 
         # Create a mapping from sequence → list of scores if multiple columns
         if len(score_col) == 1:
-            self.seq_to_score = dict(zip(df[seq_col], df[score_col[0]]))
+            self.seq_to_score = dict(zip(df["sequence"], df[score_col[0]]))
         else:
-            self.seq_to_score = dict(zip(df[seq_col], df[score_col].values.tolist()))
+            self.seq_to_score = dict(zip(df["sequence"], df[score_col].values.tolist()))
 
         print("Finished preparing data!")