Ensure editing rate is calculated by finding bulk samples from the us…

…er-specified condition column

bean/framework/ReporterScreen.py

@@ -483,6 +483,7 @@ def get_guide_edit_rate(
         return_result=False,
         count_layer="X_bcmatch",
         edit_layer="edits",
+        condition_col = "condition",
         unsorted_condition_label=None,
     ):
         """
@@ -506,10 +507,12 @@ def get_guide_edit_rate(
             )
         if unsorted_condition_label is not None:
             bulk_idx = np.where(
-                self.samples.index.astype(str).map(
+                self.samples[condition_col].astype(str).map(
                     lambda s: unsorted_condition_label in s
                 )
             )[0]
+            if len(bulk_idx) == 0:
+                raise ValueError(f"{unsorted_condition_label} is not found in ReporterScreen.samples['{condition_col}]. Check your input.")
         else:
             bulk_idx = np.arange(0, len(self.samples)).astype(int)
 

bean/notebooks/sample_quality_report.ipynb

@@ -305,6 +305,7 @@
     "    bdata.get_guide_edit_rate(\n",
     "        editable_base_start=edit_quantification_start_pos,\n",
     "        editable_base_end=edit_quantification_end_pos,\n",
+    "        condition_col=condition_label,\n",
     "        unsorted_condition_label=ctrl_cond,\n",
     "    )\n",
     "    be.qc.plot_guide_edit_rates(bdata)"