flattened allele data for plot

edilytics · Mar 20, 2024 · 33f7e24 · 33f7e24
1 parent 7bbebc0
commit 33f7e24
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CRISPResso2/CRISPRessoPlot.py b/CRISPResso2/CRISPRessoPlot.py
@@ -2560,7 +2560,7 @@ def _annotate_heatmap(self, ax, mesh):
 
 
         for x, y, m, color, val, per_element_dict  in zip(xpos.flat, ypos.flat,
-                                       mesh.get_array(), mesh.get_facecolors(),
+                                       mesh.get_array().flat, mesh.get_facecolors(),
                                        self.annot_data.flat, self.per_element_annot_kws.flat):
             #print per_element_dict
             if m is not np.ma.masked: