Pass protocol to do_subtraction_plot

CardiacModelling · Jun 28, 2024 · 8d9cc2e · 8d9cc2e
1 parent 7766770
commit 8d9cc2e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/scripts/run_herg_qc.py b/scripts/run_herg_qc.py
@@ -775,11 +775,10 @@ def extract_protocol(readname, savename, time_strs, selected_wells, args):
             continue
 
         sweeps = sorted(list(sub_df.sweep.unique()))
-        sub_df = sub_df.set_index('sweep')
-        logging.debug(sub_df)
 
         do_subtraction_plot(fig, times, sweeps, before_current, after_current,
-                            extract_df, voltages, ramp_bounds, well=well)
+                            extract_df, voltages, ramp_bounds, well=well,
+                            protocol=savename)
 
         fig.savefig(os.path.join(subtraction_plots_dir,
                                  f"{saveID}-{savename}-{well}-sweep{sweep}-subtraction"))